Upload app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from indexer import index_text, answer_query
 
 # Gradio interface function to handle the RAG system
-def rag_system(query):
+def rag_system(query,history):
     # Index the input text
     vectorstore = index_text()
     
@@ -11,14 +11,14 @@ def rag_system(query):
     
     return answer
 
-# Build the Gradio interface
-iface = gr.Interface(
-    fn=rag_system, 
-    inputs=["text"], 
-    outputs="text", 
-    title="AI Alignment Bot",
-    description="If the model can't answer, it will say sorry I don't know"
+
+iface = gr.ChatInterface(
+    rag_system,
+    type="messages",
+    title="AI Alignment ChatBot",
+    description="Let's understand AI Alignment",
 )
 
+
 # Launch the app
 iface.launch()