Update app.py

app.py

@@ -82,8 +82,8 @@ def search_chroma(query, n_results=5):
     )
     return results["documents"][0]
 
-def generate_response(user_query: str) -> str:
-    docs = search_chroma(user_query)
+def generate_response(message, history=[]):
+    docs = search_chroma(message)
     context_str = "\n\n".join(docs)
 
     sys_prompt = """You are a helpful assistant. Based on the context and user query, provide an accurate and concise answer.
@@ -94,7 +94,7 @@ def generate_response(user_query: str) -> str:
     - Do not hallucinate or make up answers.
     - Keep the tone friendly."""
 
-    prompt = f"{sys_prompt}\n\ncontext:\n{context_str}\n\nQuestion: {user_query}\nAnswer:"
+    prompt = f"{sys_prompt}\n\ncontext:\n{context_str}\n\nQuestion: {message}\nAnswer:"
 
     response = llm(
         prompt,
@@ -104,6 +104,7 @@ def generate_response(user_query: str) -> str:
         stop=["Q:", "\n"],
         echo=True
     )
+
     return response["choices"][0]["text"].split("Answer:")[-1].strip()
 
 # ---------------------- Gradio Interface ----------------------
@@ -116,7 +117,7 @@ def gradio_chat_interface(message, history):
 demo = gr.ChatInterface(
     fn=generate_response,
     title="🦙 LLaMA-3 FAQ Chatbot",
-    chatbot=gr.Chatbot(label="Ask me anything about Jupiter Money!"),
+    chatbot=gr.Chatbot(label="Ask me anything about Jupiter Money!", type="messages"),
     examples=[
         "What is Jupiter Edge credit card?",
         "What happens if I miss a payment?",