Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
+from fastapi import FastAPI
+import json
 
 # Load chatbot model
 print("Loading DialoGPT model...")
@@ -35,7 +37,7 @@ def chatbot_response(message, history=None, session_id="default"):
     
     return response, emotion, score, chat_histories[session_id]
 
-def api_chatbot_response(message, session_id="default"):
+def api_chatbot_response(message: str, session_id: str = "default"):
     """API endpoint version that returns a structured response"""
     response, emotion, score, _ = chatbot_response(message, None, session_id)
     
@@ -46,102 +48,64 @@ def api_chatbot_response(message, session_id="default"):
         "session_id": session_id
     }
 
-def get_chat_history(session_id="default"):
+def get_chat_history(session_id: str = "default"):
     """Get chat history for a specific session"""
     if session_id in chat_histories:
         return chat_histories[session_id]
     return []
 
-def clear_history(session_id="default"):
+def clear_history(session_id: str = "default"):
     """Clear chat history for a specific session"""
     if session_id in chat_histories:
         chat_histories[session_id] = []
-        return f"History cleared for session {session_id}"
-    return f"Session {session_id} not found"
+        return {"status": "success", "message": f"History cleared for session {session_id}"}
+    return {"status": "error", "message": f"Session {session_id} not found"}
 
-# Define UI interface
-with gr.Blocks(title="Mental Health Chatbot") as ui_interface:
-    gr.Markdown("# 🧠 Mental Health Chatbot")
-    
-    with gr.Row():
-        with gr.Column(scale=3):
-            chatbot = gr.Chatbot(height=400, label="Conversation")
-            
-            with gr.Row():
-                message = gr.Textbox(placeholder="Type your message here...", label="You", show_label=False)
-                submit_btn = gr.Button("Send")
-            
-            with gr.Row():
-                session_id = gr.Textbox(value="default", label="Session ID")
-                clear_btn = gr.Button("Clear Chat")
-        
-        with gr.Column(scale=1):
-            emotion_label = gr.Textbox(label="Emotion Detected")
-            emotion_score = gr.Number(label="Confidence Score")
-    
-    # Set up event handlers
-    def respond(message, chat_history, session_id):
-        response, emotion, score, _ = chatbot_response(message, chat_history, session_id)
-        chat_history.append((message, response))
-        return "", chat_history, emotion, score
+# Create FastAPI app
+app = FastAPI()
+
+# Create Gradio app
+with gr.Blocks() as gradio_interface:
+    gr.Markdown("# API Documentation")
+    gr.Markdown("Use these endpoints in your React Native app:")
     
-    submit_btn.click(
-        respond, 
-        [message, chatbot, session_id], 
-        [message, chatbot, emotion_label, emotion_score]
-    )
+    with gr.Accordion("Chat Endpoint"):
+        gr.Markdown("""
+        **POST /chat**
+        - Parameters: `message`, `session_id` (optional)
+        - Returns: JSON with bot_response, emotion, emotion_score, session_id
+        """)
     
-    message.submit(
-        respond, 
-        [message, chatbot, session_id], 
-        [message, chatbot, emotion_label, emotion_score]
-    )
+    with gr.Accordion("History Endpoint"):
+        gr.Markdown("""
+        **GET /history**
+        - Parameters: `session_id` (optional)
+        - Returns: JSON array of message/response pairs
+        """)
     
-    clear_btn.click(
-        lambda s: ([], clear_history(s), "", 0), 
-        [session_id], 
-        [chatbot, emotion_label, emotion_score]
-    )
+    with gr.Accordion("Clear History Endpoint"):
+        gr.Markdown("""
+        **POST /clear_history**
+        - Parameters: `session_id` (optional)
+        - Returns: JSON with status and message
+        """)
 
-# Define API interface
-api_interface = gr.Interface(
-    fn=api_chatbot_response,
-    inputs=[
-        gr.Textbox(label="Message"),
-        gr.Textbox(label="Session ID", value="default")
-    ],
-    outputs=gr.JSON(label="Response"),
-    title="Mental Health Chatbot API",
-    description="Send a message to get chatbot response with emotion analysis",
-    examples=[
-        ["I'm feeling sad today", "user1"],
-        ["I'm so excited about my new job!", "user2"],
-        ["I'm worried about my exam tomorrow", "user3"]
-    ]
-)
+# Mount Gradio interface
+gradio_app = gr.mount_gradio_app(app, gradio_interface, path="/")
 
-history_api = gr.Interface(
-    fn=get_chat_history,
-    inputs=gr.Textbox(label="Session ID", value="default"),
-    outputs=gr.JSON(label="Chat History"),
-    title="Chat History API",
-    description="Get chat history for a specific session"
-)
+# Add API endpoints
+@app.post("/chat")
+async def chat_endpoint(message: str, session_id: str = "default"):
+    return api_chatbot_response(message, session_id)
 
-clear_api = gr.Interface(
-    fn=clear_history,
-    inputs=gr.Textbox(label="Session ID", value="default"),
-    outputs=gr.Textbox(label="Result"),
-    title="Clear History API",
-    description="Clear chat history for a specific session"
-)
+@app.get("/history")
+async def history_endpoint(session_id: str = "default"):
+    return get_chat_history(session_id)
 
-# Combine all interfaces
-demo = gr.TabbedInterface(
-    [ui_interface, api_interface, history_api, clear_api], 
-    ["Chat UI", "Chat API", "History API", "Clear API"]
-)
+@app.post("/clear_history")
+async def clear_endpoint(session_id: str = "default"):
+    return clear_history(session_id)
 
-# Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)