added correct gradio code

app.py

@@ -1,56 +1,36 @@
 import os
-import traceback
 import requests
-from flask import Flask, request, jsonify
-from flask_cors import CORS
+import traceback
+import gradio as gr
 from dotenv import load_dotenv
 
-# Load environment variables (from Hugging Face Space secrets)
+# Load secrets (from Hugging Face Spaces "Variables and secrets")
 load_dotenv()
 
 # Configuration
 OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
-OPENROUTER_MODEL = os.getenv("OPENROUTER_MODEL", "openai/gpt-4.1-mini")
+OPENROUTER_MODEL = os.getenv("OPENROUTER_MODEL")
 APP_TITLE = os.getenv("APP_TITLE", "Selva Chat Bot")
 OPENROUTER_URL = "https://openrouter.ai/api/v1/chat/completions"
 
-# Create Flask app
-app = Flask(__name__)
-
-# Enable CORS for all routes
-CORS(app, origins=["*"])
 
-@app.route("/")
-def root():
-    return jsonify({"message": "Selva Chat Bot API", "status": "running"})
-
-@app.route("/health")
-def health():
-    redacted = OPENROUTER_API_KEY[:8] + "..." if OPENROUTER_API_KEY else ""
-    return jsonify({"ok": True, "model": OPENROUTER_MODEL, "key": redacted})
-
-@app.route("/api/chat", methods=["POST"])
-def chat():
+def chatbot_fn(message, history):
+    """
+    message: latest user input (string)
+    history: list of tuples (user, bot)
+    """
     try:
-        payload = request.get_json(silent=True) or {}
-        user_message = (payload.get("message") or "").strip()
-        history = payload.get("history") or []
+        if not OPENROUTER_API_KEY:
+            return history + [(message, "❌ API key not configured.")]
 
-        if not user_message:
-            return jsonify({"error": "Message is required"}), 400
+        # Convert Gradio history to OpenRouter messages format
+        messages = [{"role": "system", "content": "You are a helpful, concise, and friendly assistant for Selva's Chat-Bot."}]
+        for user_msg, bot_msg in history:
+            messages.append({"role": "user", "content": user_msg})
+            messages.append({"role": "assistant", "content": bot_msg})
 
-        if not OPENROUTER_API_KEY:
-            return jsonify({"error": "API key not configured"}), 500
-
-        # Build messages array
-        messages = []
-        if not any(m.get("role") == "system" for m in history):
-            messages.append({
-                "role": "system",
-                "content": "You are a helpful, concise, and friendly assistant for Selva's Chat-Bot."
-            })
-        messages.extend([m for m in history if m.get("role") in ("system", "user", "assistant")])
-        messages.append({"role": "user", "content": user_message})
+        # Add latest user query
+        messages.append({"role": "user", "content": message})
 
         # API request
         headers = {
@@ -58,20 +38,19 @@ def chat():
             "Content-Type": "application/json",
             "X-Title": APP_TITLE,
         }
-
         body = {
             "model": OPENROUTER_MODEL,
             "messages": messages,
             "temperature": 0.7,
-            "max_tokens": 1000
+            "max_tokens": 1000,
         }
 
         response = requests.post(OPENROUTER_URL, headers=headers, json=body, timeout=30)
 
         if response.status_code != 200:
             error_data = response.json() if response.headers.get('content-type') == 'application/json' else {"error": response.text}
-            error_message = error_data.get("error", {}).get("message", "API request failed") if isinstance(error_data.get("error"), dict) else str(error_data.get("error", "Unknown error"))
-            return jsonify({"error": error_message}), response.status_code
+            error_message = error_data.get("error", "API request failed")
+            return history + [(message, f"❌ Error: {error_message}")]
 
         data = response.json()
         reply = (
@@ -80,19 +59,28 @@ def chat():
             .get("content", "Sorry, I couldn't generate a response.")
         )
 
-        return jsonify({"reply": reply})
+        return history + [(message, reply)]
 
     except requests.exceptions.Timeout:
-        return jsonify({"error": "Request timeout. Please try again."}), 504
-    except requests.exceptions.RequestException as e:
-        print(f"Request error: {e}")
-        return jsonify({"error": "Failed to connect to AI service."}), 503
+        return history + [(message, "❌ Request timeout. Please try again.")]
     except Exception as e:
-        print(f"Unexpected error: {e}")
         traceback.print_exc()
-        return jsonify({"error": "Internal server error."}), 500
+        return history + [(message, f"❌ Internal error: {e}")]
+
+
+# Gradio Chatbot UI
+with gr.Blocks(title=APP_TITLE) as demo:
+    gr.Markdown(f"# 🤖 {APP_TITLE}")
+    chatbot = gr.Chatbot(height=500)
+    msg = gr.Textbox(placeholder="Type your message here...")
+    clear = gr.Button("Clear")
+
+    def user_submit(user_message, chat_history):
+        return "", chatbot_fn(user_message, chat_history)
+
+    msg.submit(user_submit, [msg, chatbot], [msg, chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
 
-# Hugging Face requires binding to 0.0.0.0 and PORT env
+# Hugging Face Spaces entrypoint
 if __name__ == "__main__":
-    port = int(os.environ.get("PORT", 7860))  # default to 7860
-    app.run(host="0.0.0.0", port=port, debug=False)
+    demo.launch(server_name="0.0.0.0", server_port=7860, debug=True, pwa=True)