Update app.py

app.py

@@ -2,11 +2,13 @@ import os
 import gradio as gr
 from dotenv import load_dotenv
 from openai import OpenAI
+from fastapi import FastAPI
+import threading
+import uvicorn
 from prompts.initial_prompt import INITIAL_PROMPT
 from prompts.main_prompt import MAIN_PROMPT
-from fastapi import FastAPI
 
-# Load OpenAI API Key from .env file
+# ✅ Load OpenAI API Key
 if os.path.exists(".env"):
     load_dotenv(".env")
 
@@ -14,19 +16,18 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
 client = OpenAI(api_key=OPENAI_API_KEY)
 
-# ✅ Create FastAPI app
-app = FastAPI()
+# ✅ FastAPI App for Serving Prompts
+fastapi_app = FastAPI()
 
-# ✅ API Endpoints to Serve Prompts
-@app.get("/initial_prompt")
+@fastapi_app.get("/initial_prompt")
 def get_initial_prompt():
     return {"prompt": INITIAL_PROMPT}
 
-@app.get("/main_prompt")
+@fastapi_app.get("/main_prompt")
 def get_main_prompt():
     return {"prompt": MAIN_PROMPT}
 
-# ✅ Chatbot function
+# ✅ Chatbot Function
 def gpt_call(history, user_message,
              model="gpt-4o-mini",
              max_tokens=512,
@@ -64,12 +65,14 @@ def respond(user_message, history):
 
     return "", history
 
-with gr.Blocks() as demo:
+# ✅ Gradio UI
+with gr.Blocks() as gradio_app:
     gr.Markdown("## Simple Chat Interface")
 
     chatbot = gr.Chatbot(
-        value=[("", INITIAL_PROMPT)],  
-        height=500
+        value=[{"role": "assistant", "content": INITIAL_PROMPT}],
+        height=500,
+        type="messages"
     )
 
     state_history = gr.State([("", INITIAL_PROMPT)])
@@ -89,10 +92,11 @@ with gr.Blocks() as demo:
         outputs=[state_history]
     )
 
-# ✅ Launch Gradio and FastAPI together
-import threading
-
+# ✅ Run FastAPI and Gradio Together
 def run_gradio():
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
+    gradio_app.launch(server_name="0.0.0.0", server_port=7860)
 
 threading.Thread(target=run_gradio).start()
+
+if __name__ == "__main__":
+    uvicorn.run(fastapi_app, host="0.0.0.0", port=8000)