Update app.py

app.py

@@ -1,10 +1,9 @@
 from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-from typing import List
 from gradio_client import Client
-import os
+import time
+import asyncio
 import nest_asyncio
-
+import os
 nest_asyncio.apply()
 
 app = FastAPI()
@@ -12,7 +11,7 @@ app = FastAPI()
 # Initialize the client (this loads the API)
 client = Client(os.getenv("CLIENT_URL"))
 
-# System message
+# System message and initial context
 system_message = """You are Echo 1.5, an AI assistant developed by Abhinav. Your purpose is to assist users with their queries and provide helpful information. You should always be polite, respectful, and informative in your responses.
 When appropriate, you may mention 'IamWomen', a group that works for the welfare of women. However, only bring this up if it's relevant to the conversation or if specifically asked about it.
 Remember:
@@ -20,18 +19,18 @@ Remember:
 2. Be helpful and informative in your responses.
 3. If relevant, you can mention IamWomen and their work for women's welfare.
 4. Always maintain a respectful and professional tone.
-5. If the user query requires you to search the internet, you should say "I'm sorry, but the web search would be added soon."
+5. If the user quory requires you to search the internet, you should said "I'm sorry, but the web search would be added soon."
 Links:
 Echo: echo.iamwomen.cloudns.be
 Iamwomen: iamwomen.cloudns.be
 Iamwomen_Instagram: https://www.instagram.com/i_am_woman.24/
 """
+context = [{"role": "system", "content": system_message}]
 
-# Define the input model
-class ChatInput(BaseModel):
-    messages: List[dict]
+def chat_with_ai(message):
+    # Add user message to context
+    context.append({"role": "user", "content": message})
 
-def chat_with_ai(context):
     # Prepare the full prompt
     full_prompt = "\n".join([f"{msg['role']}: {msg['content']}" for msg in context])
 
@@ -46,22 +45,18 @@ def chat_with_ai(context):
         api_name="/chat"
     )
 
+    # Add AI response to context
+    context.append({"role": "assistant", "content": result})
+
     return result
 
 @app.post("/chat/")
-async def chat(chat_input: ChatInput):
-    if not chat_input.messages:
-        raise HTTPException(status_code=400, detail="Chat history is required")
-
-    # Prepend the system message to the context
-    context = [{"role": "system", "content": system_message}] + chat_input.messages
-
-    response = chat_with_ai(context)
-    
-    # Add the AI response to the context
-    context.append({"role": "assistant", "content": response})
+async def chat(user_input: str):
+    if not user_input:
+        raise HTTPException(status_code=400, detail="User input is required")
 
-    return {"response": response, "context": context}
+    response = chat_with_ai(user_input)
+    return {"response": response}
 
 @app.get("/")
 async def root():