Update app.py

app.py

@@ -6,25 +6,17 @@ from threading import Thread
 import re
 import uuid
 from openai import OpenAI
-import requests
-from io import BytesIO
-import os
-#from PIL import Image
 
 client = OpenAI(
     base_url="https://a7g1ajqixo23revq.us-east-1.aws.endpoints.huggingface.cloud/v1/",
     api_key="hf_XXXXX"
 )
 
-from PIL import Image
-from pix2tex import cli as pix2tex
-# Load model once to avoid re-init every time
-ocr_model = pix2tex.LatexOCR()
-def image_to_latex(image_path):
-    image = Image.open(image_path).convert("RGB")
-    return ocr_model(image)
+def format_math(text):
+    text = re.sub(r"\[(.*?)\]", r"$$\1$$", text, flags=re.DOTALL)
+    text = text.replace(r"\(", "$").replace(r"\)", "$")
+    return text
 
-    
 # Global dictionary to store all conversations: {id: {"title": str, "messages": list}}
 conversations = {}
 
@@ -43,11 +35,7 @@ def generate_response(user_message,
     if not user_message.strip():
         return history_state, history_state
 
-    system_message =  (
-    "You are Ramanujan Ganit R1, a state-of-the-art math reasoning assistant created by Fractal AI Research. "
-    "You specialize in solving complex mathematics and science problems with step-by-step reasoning. "
-    "You are *not* ChatGPT, OpenAI, or DeepSeek. If asked about your identity, always say you are Ramanujan Ganit R1 developed by Fractal AI Research."
-)
+    system_message = "Your role as an assistant..."
     messages = [{"role": "system", "content": system_message}]
     for m in history_state:
         messages.append({"role": m["role"], "content": m["content"]})
@@ -162,19 +150,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=4):
             #chatbot = gr.Chatbot(label="Chat", type="messages")
             chatbot = gr.Chatbot(label="Chat", type="messages", height=520)
-            
             with gr.Row():
                 user_input = gr.Textbox(label="User Input", placeholder="Type your question here...", lines=3, scale=8)
                 with gr.Column():
                     submit_button = gr.Button("Send", variant="primary", scale=1)
                     clear_button = gr.Button("Clear", scale=1)
-            
-            with gr.Row():
-                image_input = gr.Image(type="filepath", label="Upload Image")
-                output_text = gr.Textbox(label="LaTeX Output")
-            image_input.change(fn=image_to_latex, inputs=image_input, outputs=output_text)
-            
-                    
             gr.Markdown("**Try these examples:**")
             with gr.Row():
                 example1_button = gr.Button("JEE Main 2025\nCombinatorics")