Update app.py

app.py

@@ -125,9 +125,9 @@ def process_and_query(state, question):
             generated_text = generator_tokenizer.decode(generator_outputs[0], skip_special_tokens=True)
 
         # Update conversation history
-        state["conversation"].append({"question": question, "answer": generated_text})
+        state.append({"question": question, "answer": generated_text})
 
-        return {"message": generated_text, "conversation": state["conversation"]}
+        return {"message": generated_text, "conversation": state}
 
     return {"error": "No question provided"}
 
@@ -142,10 +142,10 @@ with gr.Blocks() as demo:
         upload_button.click(fn=upload_files, inputs=upload, outputs=upload_output)
 
     with gr.Tab("Query"):
-        state = gr.State(initial_value={"conversation": []})
         query = gr.Textbox(label="Enter your query")
         query_button = gr.Button("Search")
         query_output = gr.Textbox()
+        state = gr.State()
         query_button.click(fn=process_and_query, inputs=[state, query], outputs=query_output)
 
 demo.launch()