Update app.py

app.py

@@ -1,372 +1,215 @@
 import gradio as gr
 import requests
 from html import escape
-import re
-from typing import List, Tuple
-import time
 
-# API endpoints
-API_URL = "https://aham2api-3.onrender.com/v1/chat/completions"
-MODELS_URL = "https://aham2api-3.onrender.com/v1/models"
-
-# Enhanced CSS with modern UI
-css = """
-:root {
-    --primary: #4f46e5;
-    --primary-hover: #4338ca;
-    --background: #ffffff;
-    --text: #111827;
-    --user-bg: #f0f7ff;
-    --bot-bg: #f9fafb;
-    --border: #e5e7eb;
-    --shadow: 0 1px 3px rgba(0,0,0,0.1);
-    --radius: 12px;
-    --input-bg: #f9fafb;
-}
+# API configurations
+DEEPSEEK_API_URL = "https://typegpt-api.aham2.com/v1/chat/completions"  # TypeGPT API for deepseek-r1
+DEFAULT_API_URL = "https://aham2api-3.onrender.com/v1/chat/completions"  # Default API for other models
+
+# List of available models
+AVAILABLE_MODELS = [
+    "deepseek-r1",
+    "gpt-4o",
+    "gpt-4o-latest",
+    "chatgpt-4o-latest",
+    "gemini-1.5-pro",
+    "gemini-1.5-pro-latest",
+    "gemini-flash-2.0",
+    "gemini-1.5-flash",
+    "claude-3-5-sonnet",
+    "claude-3-5-sonnet-20240620",
+    "anthropic/claude-3.5-sonnet",
+    "mistral-large",
+    "deepseek-v3",
+    "llama-3.1-405b",
+    "Meta-Llama-3.1-405B-Instruct-Turbo",
+    "Meta-Llama-3.3-70B-Instruct-Turbo",
+    "grok-2",
+    "qwen-plus-latest",
+    "qwen-turbo-latest",
+    "dbrx-instruct",
+    "claude",
+    "qwen-2.5-32b",
+    "qwen-2.5-coder-32b",
+    "qwen-qwq-32b",
+    "gemma2-9b-it",
+    "deepseek-r1-distill-llama-70b",
+    "o3-mini",
+    "Claude-sonnet-3.7",
+    "type/deepseek-r1",
+    "samu/deepseek-r1"
+]
+
+def get_api_url(model_name):
+    """Determine which API endpoint to use based on model selection"""
+    if model_name.lower() in ["deepseek-r1", "type/deepseek-r1", "samu/deepseek-r1"]:
+        return DEEPSEEK_API_URL
+    return DEFAULT_API_URL
+
+def convert_latex(text):
+    """Convert LaTeX expressions to HTML for rendering"""
+    # Handle block equations
+    text = text.replace('$$', '</p><div class="latex-block">')
+    text = text.replace('\\[', '</p><div class="latex-block">')
+    text = text.replace('\\]', '</div><p>')
+    
+    # Handle inline equations
+    text = text.replace('$', '<span class="latex-inline">')
+    text = text.replace('\\(', '<span class="latex-inline">')
+    text = text.replace('\\)', '</span>')
+    
+    return text
 
-.dark {
-    --primary: #6366f1;
-    --primary-hover: #818cf8;
-    --background: #1a1a1a;
-    --text: #f3f4f6;
-    --user-bg: #2b3d4f;
-    --bot-bg: #2d2d2d;
-    --border: #404040;
-    --input-bg: #2d2d2d;
-}
+def postprocess_message(message):
+    """Process message for HTML display with LaTeX support"""
+    if not message:
+        return ""
+    
+    # Escape HTML first for security
+    message = escape(message)
+    
+    # Convert LaTeX expressions
+    message = convert_latex(message)
+    
+    # Handle newlines
+    message = message.replace('\n', '<br>')
+    
+    return f'<div class="message-content">{message}</div>'
 
-.gradio-container {
-    background: var(--background) !important;
-    color: var(--text) !important;
-    max-width: 900px !important;
-    margin: 0 auto;
-    font-family: 'Inter', system-ui, sans-serif;
-}
+def chat_with_aham(user_message, history, selected_model):
+    if not user_message.strip():
+        return "Please enter a message.", history
+    
+    # Prepare the conversation history for the API
+    messages = []
+    
+    # Add previous conversation history
+    for user_msg, ai_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": ai_msg})
+    
+    # Add the current user message
+    messages.append({"role": "user", "content": user_message})
 
-.chat-container {
-    height: 80vh;
-    border: 1px solid var(--border);
-    border-radius: var(--radius);
-    overflow: hidden;
-    box-shadow: var(--shadow);
-    display: flex;
-    flex-direction: column;
-}
+    payload = {
+        "model": selected_model,
+        "messages": messages,
+        "temperature": 0.7
+    }
 
-.chatbot {
-    height: 100%;
-    padding: 0 !important;
-    background: transparent !important;
-    display: flex;
-    flex-direction: column;
-    overflow-y: auto;
-}
+    headers = {"Content-Type": "application/json"}
+    api_url = get_api_url(selected_model)
 
-.chatbot .wrap {
-    padding: 20px;
-    flex-grow: 1;
-    overflow-y: auto;
-}
+    try:
+        response = requests.post(api_url, json=payload, headers=headers)
 
-.message {
-    padding: 12px 16px;
-    border-radius: var(--radius);
-    margin: 8px 0;
-    max-width: 85%;
+        if response.status_code == 200:
+            data = response.json()
+            ai_response = data.get("choices", [{}])[0].get("message", {}).get("content", "No response received.")
+            
+            # Append the current interaction to history
+            history.append((user_message, ai_response))
+            return "", history  # Clear input, return updated history
+        else:
+            error_msg = f"Error: {response.status_code}, {response.text}"
+            history.append((user_message, error_msg))
+            return "", history
+    
+    except Exception as e:
+        error_msg = f"Error: {str(e)}"
+        history.append((user_message, error_msg))
+        return "", history
+
+# Custom CSS for LaTeX rendering and chat interface
+custom_css = """
+.latex-block {
+    padding: 10px;
+    margin: 10px 0;
+    background-color: #f5f5f5;
+    border-radius: 5px;
+    overflow-x: auto;
+    text-align: center;
+    font-family: monospace;
+}
+
+.latex-inline {
+    background-color: #f0f0f0;
+    padding: 2px 4px;
+    border-radius: 3px;
+    font-family: monospace;
+}
+
+.message-content {
     line-height: 1.6;
-    font-size: 15px;
-    transition: all 0.3s ease;
-    box-shadow: var(--shadow);
-}
-
-.user-message {
-    background: var(--user-bg);
-    margin-left: auto;
-    border-bottom-right-radius: 4px;
-    color: var(--text);
-}
-
-.bot-message {
-    background: var(--bot-bg);
-    margin-right: auto;
-    border-bottom-left-radius: 4px;
-    color: var(--text);
-}
-
-.input-container {
-    padding: 16px;
-    background: var(--background);
-    border-top: 1px solid var(--border);
-    display: flex;
-    gap: 8px;
-}
-
-.textbox {
-    background: var(--input-bg) !important;
-    border-radius: var(--radius) !important;
-    border: 1px solid var(--border) !important;
-    padding: 12px 16px !important;
-}
-
-.textbox:focus {
-    border-color: var(--primary) !important;
-    box-shadow: 0 0 0 2px rgba(79, 70, 229, 0.2) !important;
-}
-
-.send-btn {
-    background: var(--primary) !important;
-    color: white !important;
-    border-radius: var(--radius) !important;
-    border: none !important;
-    padding: 0 24px !important;
 }
 
-.send-btn:hover {
-    background: var(--primary-hover) !important;
+.dark .latex-block, .dark .latex-inline {
+    background-color: #2d2d2d;
 }
 
-.dark-toggle {
-    position: absolute;
-    right: 20px;
-    top: 20px;
-    z-index: 1000;
-    background: transparent !important;
-    border: none !important;
-    box-shadow: none !important;
-    color: var(--text) !important;
-    font-size: 20px !important;
+#chatbot {
+    min-height: 500px;
 }
 
-.model-selector {
-    padding: 0 16px 16px 16px;
-    background: var(--background);
+.model-info {
+    font-size: 0.9em;
+    color: #666;
+    margin-top: 5px;
 }
-
-.controls {
-    display: flex;
-    justify-content: space-between;
-    align-items: center;
-    padding: 12px 16px;
-    border-bottom: 1px solid var(--border);
-    background: var(--background);
-}
-
-.typing-indicator {
-    display: inline-flex;
-    gap: 4px;
-    align-items: center;
-    color: #6b7280;
-    font-size: 14px;
-}
-
-.typing-dot {
-    width: 8px;
-    height: 8px;
-    background: #9ca3af;
-    border-radius: 50%;
-    animation: typing 1.4s infinite ease-in-out;
-}
-
-.typing-dot:nth-child(2) {
-    animation-delay: 0.2s;
-}
-
-.typing-dot:nth-child(3) {
-    animation-delay: 0.4s;
-}
-
-@keyframes typing {
-    0%, 60%, 100% { transform: translateY(0); }
-    30% { transform: translateY(-4px); }
-}
-
-.katex { font-size: 1.1em !important; }
 """
 
-def get_available_models() -> List[str]:
-    try:
-        response = requests.get(MODELS_URL, timeout=10)
-        if response.status_code == 200:
-            return sorted([m['id'] for m in response.json().get('data', [])])
-        return ["samura-gpt-4o", "samura-claude-3-5-sonnet"]
-    except Exception as e:
-        print(f"Error fetching models: {e}")
-        return ["samura-gpt-4o", "samura-claude-3-5-sonnet"]
-
-def chat_completion(messages: List[dict], model: str) -> dict:
-    headers = {"Content-Type": "application/json"}
-    data = {
-        "model": model,
-        "messages": messages,
-        "temperature": 0.7,
-        "max_tokens": 2500
-    }
-    try:
-        response = requests.post(API_URL, headers=headers, json=data, timeout=40)
-        response.raise_for_status()
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        return {"error": f"API request failed: {str(e)}"}
-    except Exception as e:
-        return {"error": f"Unexpected error: {str(e)}"}
-
-def format_message(text: str) -> str:
-    if not text:
-        return ""
-    text = escape(text)
-    # Handle block LaTeX ($$...$$)
-    text = re.sub(r'\$\$(.*?)\$\$', r'<div class="katex-block">$$\1$$</div>', text, flags=re.DOTALL)
-    # Handle inline LaTeX ($...$)
-    text = re.sub(r'\$(.*?)\$', r'<span class="katex-inline">$\1$</span>', text)
-    # Handle specific fraction case
-    text = text.replace(r'\frac{1}{f} = \frac{1}{v} + \frac{1}{u}', 
-                       r'<div class="katex-block">\frac{1}{f} = \frac{1}{v} + \frac{1}{u}</div>')
-    return text.replace("\n", "<br>")
-
-def create_typing_indicator():
-    return """
-    <div class="typing-indicator">
-        <div class="typing-dot"></div>
-        <div class="typing-dot"></div>
-        <div class="typing-dot"></div>
-    </div>
-    """
-
-def respond(message: str, chat_history: List[Tuple[str, str]], model: str):
-    if not message.strip():
-        yield chat_history, ""
-        return
-    
-    # Add user message
-    chat_history.append((format_message(message), None))
-    yield chat_history, ""
+# Gradio UI with ChatInterface
+with gr.Blocks(title="Aham AI Chat", css=custom_css) as demo:
+    gr.Markdown("## Chat with Aham AI (with LaTeX support)")
     
-    # Prepare API messages
-    messages = []
-    for user_msg, bot_msg in chat_history[:-1]:
-        if user_msg:
-            messages.append({"role": "user", "content": re.sub('<[^<]+?>', '', user_msg)})
-        if bot_msg:
-            messages.append({"role": "assistant", "content": re.sub('<[^<]+?>', '', bot_msg)})
+    with gr.Row():
+        model_dropdown = gr.Dropdown(
+            choices=AVAILABLE_MODELS,
+            value="deepseek-r1",
+            label="Select Model",
+            interactive=True
+        )
+        gr.Markdown("<div class='model-info'>deepseek-r1 uses TypeGPT API</div>")
     
-    # Add typing indicator
-    chat_history[-1] = (chat_history[-1][0], create_typing_indicator())
-    yield chat_history, ""
+    chatbot = gr.Chatbot(
+        height=500,
+        bubble_full_width=False,
+        render_markdown=True,
+        latex_delimiters=[
+            {"left": "$$", "right": "$$", "display": True},
+            {"left": "$", "right": "$", "display": False},
+            {"left": "\\(", "right": "\\)", "display": False},
+            {"left": "\\[", "right": "\\]", "display": True}
+        ]
+    )
     
-    # Get API response
-    response = chat_completion(messages + [{"role": "user", "content": message}], model)
+    msg = gr.Textbox(
+        label="Your Message", 
+        placeholder="Type your message here... (use $ for inline LaTeX and $$ for block equations)",
+        lines=3
+    )
     
-    if "error" in response:
-        bot_message = f"⚠️ Error: {response['error']}"
-    else:
-        bot_message = response.get("choices", [{}])[0].get("message", {}).get("content", "No response")
+    with gr.Row():
+        clear = gr.Button("Clear Conversation", variant="secondary")
+        submit = gr.Button("Send", variant="primary")
     
-    # Update chat with final response
-    chat_history[-1] = (chat_history[-1][0], format_message(bot_message))
-    yield chat_history, ""
-
-def toggle_dark_mode():
-    return """
-    <script>
-    (function() {
-        const container = document.querySelector('.gradio-container');
-        if (container) {
-            container.classList.toggle('dark');
-            localStorage.setItem('gradioDarkMode', container.classList.contains('dark'));
-        }
-    })();
-    </script>
-    """
-
-def load_dark_mode():
-    return """
-    <script>
-    (function() {
-        if (localStorage.getItem('gradioDarkMode') === 'true') {
-            document.querySelector('.gradio-container').classList.add('dark');
-        }
-    })();
-    </script>
-    """
-
-def add_math_examples():
-    examples = [
-        ["Explain the formula $E = mc^2$"],
-        ["Derive the lens formula $\frac{1}{f} = \frac{1}{v} + \frac{1}{u}$"],
-        ["Solve the quadratic equation $ax^2 + bx + c = 0$"],
-        ["What is the Pythagorean theorem?"],
-        ["Explain the concept of derivatives in calculus"],
-        ["How do I solve systems of linear equations?"]
-    ]
-    return examples
-
-with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
-    # Add initial dark mode script
-    app.load(None, None, None, _js=load_dark_mode())
-    
-    # Header with dark mode toggle
-    with gr.Column():
-        gr.Markdown("""
-        <div style="display: flex; justify-content: space-between; align-items: center;">
-            <h1 style="margin: 0;">Math Assistant 🤖</h1>
-            <button onclick="document.querySelector('.dark-toggle').click()" style="background: none; border: none; cursor: pointer; font-size: 20px;">🌓</button>
-        </div>
-        <p style="margin-top: 8px; color: var(--text); opacity: 0.8;">Ask any math question and get detailed explanations</p>
-        """)
-        
-        # Hidden dark mode toggle for JS
-        dark_toggle = gr.Button("🌓", elem_classes="dark-toggle", visible=False)
-        dark_toggle.click(None, _js=toggle_dark_mode())
+    examples = gr.Examples(
+        examples=[
+            ["What is the Pythagorean theorem?"],
+            ["Explain the formula $E = mc^2$"],
+            ["How do I solve this equation? $$\\int_0^1 x^2 dx$$"],
+            ["What's the derivative of $f(x) = \\sin(x)$?"]
+        ],
+        inputs=msg,
+        label="Example Questions (Click to try)"
+    )
     
-    with gr.Column(elem_classes="chat-container"):
-        # Controls
-        with gr.Column(elem_classes="controls"):
-            clear_btn = gr.Button("Clear Chat", size="sm")
-            model_dropdown = gr.Dropdown(
-                choices=get_available_models(),
-                value="samura-gpt-4o",
-                label="Model",
-                interactive=True,
-                scale=2
-            )
-        
-        # Chat area
-        chatbot = gr.Chatbot(
-            elem_id="chatbot",
-            bubble_full_width=False,
-            avatar_images=(
-                "https://i.imgur.com/hp3JdUq.png", 
-                "https://i.imgur.com/7WqB2Dn.png"
-            ),
-            show_label=False,
-            height="100%"
-        )
-        
-        # Input area
-        with gr.Row(elem_classes="input-container"):
-            msg = gr.Textbox(
-                placeholder="Type your math question here...",
-                lines=2,
-                max_lines=5,
-                show_label=False,
-                container=False,
-                elem_classes="textbox",
-                autofocus=True
-            )
-            submit_btn = gr.Button("Send", elem_classes="send-btn")
-        
-        # Examples
-        gr.Examples(
-            examples=add_math_examples(),
-            inputs=msg,
-            label="Try these examples:",
-            examples_per_page=3
-        )
+    def respond(message, chat_history, model):
+        return chat_with_aham(message, chat_history, model)
     
-    # Event handlers
-    msg.submit(respond, [msg, chatbot, model_dropdown], [chatbot, msg], queue=True)
-    submit_btn.click(respond, [msg, chatbot, model_dropdown], [chatbot, msg], queue=True)
-    clear_btn.click(lambda: [], None, chatbot, queue=False)
+    msg.submit(respond, [msg, chatbot, model_dropdown], [msg, chatbot])
+    submit.click(respond, [msg, chatbot, model_dropdown], [msg, chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
 
 if __name__ == "__main__":
-    app.queue(concurrency_count=5).launch()
+    demo.launch()