Update main.py

main.py

@@ -18,8 +18,8 @@ class QueryRequest(BaseModel):
     message: str
     system_prompt: str = "You are a helpful assistant capable of code generation, analysis, review, and more."
     history: Optional[List[dict]] = None
-    temperature: float = 0.7  # خفضت الـ temperature لردود أكثر دقة
-    max_new_tokens: int = 4096  # قللت القيمة لتحسين الأداء
+    temperature: float = 0.7  # خفضنا الـ temperature عشان ردود أكثر دقة
+    max_new_tokens: int = 4096  # قللنا العدد عشان نضمن ردود مكتملة
     enable_browsing: bool = False
 
 # تعريف LATEX_DELIMS
@@ -73,12 +73,14 @@ def web_search(query: str) -> str:
         google_api_key = os.getenv("GOOGLE_API_KEY")
         google_cse_id = os.getenv("GOOGLE_CSE_ID")
         if not google_api_key or not google_cse_id:
+            logger.warning("GOOGLE_API_KEY or GOOGLE_CSE_ID not set, web search disabled.")
             return "Web search requires GOOGLE_API_KEY and GOOGLE_CSE_ID to be set."
         url = f"https://www.googleapis.com/customsearch/v1?key={google_api_key}&cx={google_cse_id}&q={query}+site:mgzon.com"
         response = requests.get(url, timeout=10)
         response.raise_for_status()
         results = response.json().get("items", [])
         if not results:
+            logger.info("No web results found for query: %s", query)
             return "No web results found."
         search_results = []
         for i, item in enumerate(results[:5]):
@@ -97,10 +99,10 @@ def web_search(query: str) -> str:
             search_results.append(f"Result {i+1}:\nTitle: {title}\nLink: {link}\nContent: {page_content}\n")
         return "\n".join(search_results)
     except Exception as e:
-        logger.exception("Web search failed")
+        logger.exception("Web search failed for query: %s", query)
         return f"Web search error: {e}"
 
-# دالة request_generation
+# دالة request_generation مع logging إضافي
 @retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=4, max=10))
 def request_generation(
     api_key: str,
@@ -116,6 +118,7 @@ def request_generation(
     tool_choice: Optional[str] = None,
     deep_search: bool = False,
 ) -> Generator[str, None, None]:
+    logger.info(f"Requesting generation for model: {model_name}, endpoint: {api_base}, message: {message}")
     client = OpenAI(api_key=api_key, base_url=api_base, timeout=60.0)
     task_type = "general"
     if "code" in message.lower() or "programming" in message.lower() or any(ext in message.lower() for ext in ["python", "javascript", "react", "django", "flask"]):
@@ -151,6 +154,7 @@ def request_generation(
     tool_choice = tool_choice if tool_choice in ["auto", "none", "any", "required"] and "gpt-oss" in model_name else "none"
 
     try:
+        logger.debug(f"Sending request to {api_base} with model {model_name}")
         stream = client.chat.completions.create(
             model=model_name,
             messages=input_messages,
@@ -225,6 +229,7 @@ def request_generation(
 
         if buffer:
             yield buffer
+        logger.info(f"Generation completed successfully for model: {model_name}")
 
     except Exception as e:
         logger.exception(f"[Gateway] Streaming failed for model {model_name}: {e}")
@@ -282,6 +287,7 @@ def request_generation(
 
                 if buffer:
                     yield buffer
+                logger.info(f"Fallback generation completed successfully for model: {fallback_model}")
 
             except Exception as e2:
                 logger.exception(f"[Gateway] Streaming failed for fallback model {fallback_model}: {e2}")
@@ -318,6 +324,8 @@ def request_generation(
                             break
                     if buffer:
                         yield buffer
+                    logger.info(f"Tertiary generation completed successfully for model: {TERTIARY_MODEL_NAME}")
+
                 except Exception as e3:
                     logger.exception(f"[Gateway] Streaming failed for tertiary model {TERTIARY_MODEL_NAME}: {e3}")
                     yield f"Error: Failed to load all models: {e3}"
@@ -513,16 +521,15 @@ async def root():
         <meta charset="utf-8">
         <meta name="viewport" content="width=device-width, initial-scale=1">
         <title>MGZon Chatbot - Powered by AI</title>
-        <link href="https://unpkg.com/[email protected]/css/boxicons.min.css" rel="stylesheet">
+        <link href="https://cdn.jsdelivr.net/npm/[email protected]/css/boxicons.min.css" rel="stylesheet">
         <style>
             :root {
                 --primary-color: #1e3c72;
                 --secondary-color: #2a5298;
                 --accent-color: #ff6f61;
-                --text-color: #ffffff;
-                --card-bg: rgba(255, 255, 255, 0.1);
-                --card-hover-bg: rgba(255, 255, 255, 0.2);
-                --shadow: 0 8px 32px rgba(0, 0, 0, 0.2);
+                --text-color: #fff;
+                --container-bg: rgba(255, 255, 255, 0.1);
+                --container-border: rgba(255, 255, 255, 0.2);
             }
             body {
                 font-family: 'Arial', sans-serif;
@@ -532,31 +539,31 @@ async def root():
                 color: var(--text-color);
                 display: flex;
                 flex-direction: column;
+                align-items: center;
                 min-height: 100vh;
                 overflow-x: hidden;
             }
             .container {
-                max-width: 1200px;
-                margin: auto;
-                padding: 40px 20px;
+                max-width: 900px;
                 text-align: center;
-                background: var(--card-bg);
+                padding: 40px;
+                background: var(--container-bg);
                 border-radius: 15px;
-                box-shadow: var(--shadow);
+                box-shadow: 0 8px 32px rgba(0, 0, 0, 0.2);
                 backdrop-filter: blur(10px);
+                margin: 20px;
             }
             h1 {
                 font-size: 3.5rem;
                 margin-bottom: 20px;
                 text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.3);
-                animation: fadeIn 1s ease-in-out;
             }
             p {
                 font-size: 1.2rem;
                 line-height: 1.6;
                 margin-bottom: 30px;
             }
-            a#chatbot-link {
+            a.cta-button {
                 display: inline-block;
                 padding: 15px 40px;
                 background: var(--accent-color);
@@ -564,9 +571,10 @@ async def root():
                 text-decoration: none;
                 border-radius: 25px;
                 font-size: 1.2rem;
+                font-weight: bold;
                 transition: background 0.3s, transform 0.2s;
             }
-            a#chatbot-link:hover {
+            a.cta-button:hover {
                 background: #e55a50;
                 transform: scale(1.05);
             }
@@ -576,109 +584,89 @@ async def root():
             }
             .features h2, .integration h2 {
                 font-size: 2rem;
-                margin-bottom: 15px;
-                color: var(--text-color);
+                margin-bottom: 20px;
+                text-align: center;
             }
             .features ul {
-                background: var(--card-bg);
+                list-style: none;
+                padding: 0;
+                background: rgba(0, 0, 0, 0.2);
                 padding: 20px;
                 border-radius: 10px;
-                list-style: none;
             }
             .features li {
-                margin-bottom: 10px;
+                margin-bottom: 15px;
                 font-size: 1.1rem;
                 display: flex;
                 align-items: center;
-                gap: 10px;
+            }
+            .features li::before {
+                content: '🚀';
+                margin-right: 10px;
             }
             .integration pre {
-                background: var(--card-bg);
+                background: rgba(0, 0, 0, 0.3);
                 padding: 20px;
                 border-radius: 10px;
                 font-family: 'Courier New', monospace;
                 color: #c9e4ca;
-                white-space: pre-wrap;
-                font-size: 0.95rem;
+                overflow-x: auto;
             }
             .footer {
-                background: linear-gradient(270deg, #1f2937, #111827, #1f2937);
-                background-size: 200% 200%;
-                animation: gradient 15s ease infinite;
-                padding: 40px 20px;
-                margin-top: auto;
                 text-align: center;
-            }
-            @keyframes gradient {
-                0% { background-position: 0% 50%; }
-                50% { background-position: 100% 50%; }
-                100% { background-position: 0% 50%; }
-            }
-            .footer-card {
-                background: var(--card-bg);
                 padding: 20px;
-                border-radius: 10px;
-                box-shadow: var(--shadow);
-                transition: transform 0.3s ease, box-shadow 0.3s ease, background 0.3s ease;
-            }
-            .footer-card:hover {
-                transform: scale(1.05);
-                background: var(--card-hover-bg);
-                box-shadow: 0 10px 20px rgba(0, 0, 0, 0.3);
-            }
-            .modal {
-                display: none;
-                position: fixed;
-                z-index: 1000;
-                left: 0;
-                top: 0;
-                width: 100%;
-                height: 100%;
                 background: rgba(0, 0, 0, 0.5);
-                align-items: center;
-                justify-content: center;
+                width: 100%;
+                margin-top: 40px;
             }
-            .modal-content {
-                background: #fff;
-                padding: 20px;
-                border-radius: 10px;
-                max-width: 600px;
-                width: 90%;
-                color: #333;
+            .footer a {
+                color: var(--accent-color);
+                text-decoration: none;
+                margin: 0 10px;
             }
-            .close-btn {
-                float: right;
-                font-size: 1.5rem;
-                cursor: pointer;
+            .footer a:hover {
+                text-decoration: underline;
             }
             @keyframes fadeIn {
                 from { opacity: 0; transform: translateY(20px); }
                 to { opacity: 1; transform: translateY(0); }
             }
-            .container > *, .footer > * {
+            .container > * {
                 animation: fadeIn 0.5s ease-in-out;
             }
+            @keyframes gradient {
+                0% { background-position: 0% 50%; }
+                50% { background-position: 100% 50%; }
+                100% { background-position: 0% 50%; }
+            }
+            body {
+                background-size: 200% 200%;
+                animation: gradient 15s ease infinite;
+            }
         </style>
     </head>
     <body>
         <div class="container">
             <h1>Welcome to MGZon Chatbot 🚀</h1>
             <p>
-                MGZon Chatbot is your AI-powered assistant for code generation, analysis, and MGZon-specific queries. Built with Gradio and FastAPI, it supports multiple frameworks and languages. Ready to explore?
+                MGZon Chatbot is your AI-powered assistant for code generation, analysis, and MGZon-specific queries. Built with Gradio and FastAPI, it supports multiple frameworks (React, Django, Flask) and languages (Python, JavaScript, and more). Ready to revolutionize your projects?
             </p>
-            <a href="/gradio" id="chatbot-link">Launch Chatbot</a>
+            <a href="/gradio" class="cta-button">Launch Chatbot</a>
             <div class="features">
                 <h2>Features</h2>
                 <ul>
-                    <li><i class='bx bx-code-alt'></i> Generate code for React, Django, Flask, and more.</li>
-                    <li><i class='bx bx-analyse'></i> Analyze and review code or data with detailed insights.</li>
-                    <li><i class='bx bx-globe'></i> Web search integration for MGZon-related queries.</li>
-                    <li><i class='bx bx-bot'></i> Powered by GPT-OSS-20B and fine-tuned MGZon/Veltrix model.</li>
+                    <li>💻 Generate clean, well-commented code for React, Django, Flask, and more.</li>
+                    <li>🔍 Analyze code or data with step-by-step reasoning and insights.</li>
+                    <li>🌐 Web search integration for real-time MGZon-related information.</li>
+                    <li>🤖 Powered by GPT-OSS-20B and fine-tuned MGZon/Veltrix model.</li>
+                    <li>📝 Review and improve your code with detailed suggestions.</li>
                 </ul>
             </div>
             <div class="integration">
                 <h2>Integrate with MGZon Chatbot</h2>
-                <p>Our API supports integration with various projects, frameworks (React, Django, Flask), and languages (Python, JavaScript, etc.). Below are examples to get started:</p>
+                <p>
+                    Integrate our API into your projects with ease. Below are examples for Python, JavaScript, and cURL, supporting frameworks like React, Django, Flask, and more. Use it for web apps, mobile apps, or CLI tools.
+                </p>
                 <pre>
 # Python Example (using gradio_client)
 from gradio_client import Client
@@ -704,131 +692,68 @@ fetch('https://mgzon-mgzon-app.hf.space/api/chat', {
     })
 }).then(response => response.json()).then(data => console.log(data));
 
-# Bash Example (using curl)
+// Bash Example (using cURL)
 curl -X POST https://mgzon-mgzon-app.hf.space/api/chat \
 -H "Content-Type: application/json" \
--d '{"message":"Generate a Django model for an e-commerce product","system_prompt":"You are a coding expert","temperature":0.7,"max_new_tokens":4096}'
-
-# Ruby Example
-require 'httparty'
-response = HTTParty.post('https://mgzon-mgzon-app.hf.space/api/chat',
-  headers: { 'Content-Type' => 'application/json' },
-  body: {
-    message: 'Generate a Ruby on Rails controller for user management',
-    system_prompt: 'You are a coding expert',
-    temperature: 0.7,
-    max_new_tokens: 4096
-  }.to_json
-)
-puts response.parsed_response
-
-# PHP Example
-<?php
-$ch = curl_init('https://mgzon-mgzon-app.hf.space/api/chat');
-curl_setopt($ch, CURLOPT_POST, true);
-curl_setopt($ch, CURLOPT_HTTPHEADER, ['Content-Type: application/json']);
-curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode([
-    'message' => 'Generate a Laravel route for API authentication',
-    'system_prompt' => 'You are a coding expert',
-    'temperature' => 0.7,
-    'max_new_tokens' => 4096
-]));
-curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
-$response = curl_exec($ch);
-curl_close($ch);
-echo $response;
-?>
+-d '{"message":"Analyze the performance of a Django REST API","system_prompt":"You are an expert analyst","temperature":0.7,"max_new_tokens":4096}'
+
+# Django Integration Example
+# Add to your Django views.py
+import requests
+def chatbot_view(request):
+    response = requests.post(
+        'https://mgzon-mgzon-app.hf.space/api/chat',
+        json={
+            'message': 'Create a Django model for an e-commerce product',
+            'system_prompt': 'You are a Django expert',
+            'temperature': 0.7,
+            'max_new_tokens': 4096
+        }
+    )
+    return HttpResponse(response.json()['response'])
+
+# React Integration Example
+import React, { useState } from 'react';
+function ChatbotComponent() {
+    const [response, setResponse] = useState('');
+    const fetchResponse = async () => {
+        const res = await fetch('https://mgzon-mgzon-app.hf.space/api/chat', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                message: 'Generate a React component for a dashboard',
+                system_prompt: 'You are a React expert',
+                temperature: 0.7,
+                max_new_tokens: 4096
+            })
+        });
+        const data = await res.json();
+        setResponse(data.response);
+    };
+    return (
+        <div>
+            <button onClick={fetchResponse}>Generate Code</button>
+            <pre>{response}</pre>
+        </div>
+    );
+}
                 </pre>
-                <p>Check out our full <a href="https://mgzon.com/docs" target="_blank">API Documentation</a> for more details on endpoints, parameters, and authentication (OAuth 2.0).</p>
             </div>
         </div>
         <footer class="footer">
-            <div class="container">
-                <img src="https://raw.githubusercontent.com/Mark-Lasfar/MGZon/9a1b2149507ae61fec3bb7fb86d8d16c11852f3b/public/icons/mg.svg" alt="MGZon Logo" class="w-24 h-24 mx-auto mb-4">
-                <p class="text-gray-300 max-w-2xl mx-auto text-lg">MGZon is a leading platform for e-commerce integrations and API solutions.</p>
-                <h3 class="text-3xl font-bold mb-8 text-center">Contact Information</h3>
-                <div class="grid grid-cols-1 sm:grid-cols-2 lg:grid-cols-3 gap-6">
-                    <div class="footer-card" data-details="Reach our support team at support@mgzon.com. We aim to respond within 24 hours.">
-                        <div class="flex justify-center mb-4"><i class='bx bx-mail-send text-4xl text-blue-400'></i></div>
-                        <h4 class="text-xl font-semibold text-center">Email Us</h4>
-                        <p class="text-gray-300 text-center mt-2">Reach out to our support team via email.</p>
-                    </div>
-                    <div class="footer-card" data-details="Call +1-800-123-4567 for immediate support (Monday-Friday, 9 AM-5 PM EST).">
-                        <div class="flex justify-center mb-4"><i class='bx bx-phone text-4xl text-blue-400'></i></div>
-                        <h4 class="text-xl font-semibold text-center">Phone Support</h4>
-                        <p class="text-gray-300 text-center mt-2">Call us for immediate assistance.</p>
-                    </div>
-                    <div class="footer-card" data-details="Support available Monday-Friday, 9 AM-5 PM EST. Enterprise clients can request 24/7 support.">
-                        <div class="flex justify-center mb-4"><i class='bx bx-group text-4xl text-blue-400'></i></div>
-                        <h4 class="text-xl font-semibold text-center">Support Hours</h4>
-                        <p class="text-gray-300 text-center mt-2">Available 9 AM - 5 PM, Monday to Friday.</p>
-                    </div>
-                </div>
-                <h3 class="text-3xl font-bold mb-8 text-center">Resources</h3>
-                <div class="grid grid-cols-1 sm:grid-cols-2 lg:grid-cols-3 gap-6">
-                    <div class="footer-card" data-details="Explore API endpoints, OAuth 2.0, and integration guides at mgzon.com/developers.">
-                        <div class="flex justify-center mb-4"><i class='bx bx-code-alt text-4xl text-green-400'></i></div>
-                        <h4 class="text-xl font-semibold text-center">API Documentation</h4>
-                        <p class="text-gray-300 text-center mt-2">Explore our API endpoints and integration guides.</p>
-                    </div>
-                    <div class="footer-card" data-details="Find answers to common questions at mgzon.com/community.">
-                        <div class="flex justify-center mb-4"><i class='bx bx-help-circle text-4xl text-green-400'></i></div>
-                        <h4 class="text-xl font-semibold text-center">FAQ</h4>
-                        <p class="text-gray-300 text-center mt-2">Find answers to common questions.</p>
-                    </div>
-                    <div class="footer-card" data-details="Step-by-step guides on using MGZon at mgzon.com/docs.">
-                        <div class="flex justify-center mb-4"><i class='bx bx-book text-4xl text-green-400'></i></div>
-                        <h4 class="text-xl font-semibold text-center">Documentation</h4>
-                        <p class="text-gray-300 text-center mt-2">Learn how to use MGZon with our detailed guides.</p>
-                    </div>
-                </div>
-                <div class="mt-12 text-center">
-                    <div class="flex justify-center space-x-6 mb-6">
-                        <a href="https://github.com/Mark-Lasfar/MGZon" class="text-gray-300 hover:text-blue-400"><i class='bx bxl-github text-3xl'></i></a>
-                        <a href="https://x.com/MGZon" class="text-gray-300 hover:text-blue-400"><i class='bx bxl-twitter text-3xl'></i></a>
-                        <a href="https://www.facebook.com/people/Mark-Al-Asfar/pfbid02GMisUQ8AqWkNZjoKtWFHH1tbdHuVscN1cjcFnZWy9HkRaAsmanBfT6mhySAyqpg4l/" class="text-gray-300 hover:text-blue-400"><i class='bx bxl-facebook text-3xl'></i></a>
-                    </div>
-                    <p class="text-gray-300">© 2025 Mark Al-Asfar & MGZon AI. All rights reserved.</p>
-                </div>
-            </div>
-            <div id="footer-modal" class="modal">
-                <div class="modal-content">
-                    <span id="close-btn" class="close-btn">&times;</span>
-                    <h3 id="modal-title" class="text-2xl font-bold mb-4"></h3>
-                    <p id="modal-details" class="text-gray-700"></p>
-                </div>
-            </div>
+            <p>© 2025 MGZon AI. All rights reserved.</p>
+            <p>
+                <a href="https://mgzon.com/about">About</a> |
+                <a href="https://mgzon.com/support">Support</a> |
+                <a href="https://mgzon.com/docs">Documentation</a> |
+                <a href="https://github.com/Mark-Lasfar/MGZon">GitHub</a>
+            </p>
         </footer>
         <script>
-            // Modal functionality
-            const modal = document.getElementById('footer-modal');
-            const modalTitle = document.getElementById('modal-title');
-            const modalDetails = document.getElementById('modal-details');
-            const closeBtn = document.getElementById('close-btn');
-            const cards = document.querySelectorAll('.footer-card');
-
-            cards.forEach(card => {
-                card.addEventListener('click', () => {
-                    modalTitle.textContent = card.querySelector('h4').textContent;
-                    modalDetails.textContent = card.getAttribute('data-details');
-                    modal.style.display = 'flex';
-                });
-            });
-
-            closeBtn.addEventListener('click', () => {
-                modal.style.display = 'none';
-            });
-
-            window.addEventListener('click', (e) => {
-                if (e.target === modal) {
-                    modal.style.display = 'none';
-                }
-            });
-
-            // Redirect to /gradio with proper URL
-            document.getElementById('chatbot-link').addEventListener('click', (e) => {
+            // إصلاح الـ redirect
+            document.querySelector('.cta-button').addEventListener('click', (e) => {
                 e.preventDefault();
-                window.location.href = 'https://mgzon-mgzon-app.hf.space/gradio';
+                window.location.href = window.location.origin + '/gradio/';
             });
         </script>
     </body>