Updated conversation history to be a session state

app.py

@@ -4,10 +4,7 @@ import os
 
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-conversation_history = []
-
-def get_chatbot_response(user_message, country, language):
-    global conversation_history
+def get_chatbot_response(user_message, country, language, conversation_history):
 
     system_message = (
         f"You are a lawyer specializing in providing concise and accurate legal information based on the laws in {country}. "
@@ -40,7 +37,12 @@ def get_chatbot_response(user_message, country, language):
 
     conversation_history.append({"role": "assistant", "content": response})
 
-    return [(msg["content"], conversation_history[i + 1]["content"]) for i, msg in enumerate(conversation_history[:-1]) if msg["role"] == "user"]
+    chat_display = [
+        (msg["content"], conversation_history[i + 1]["content"]) 
+        for i, msg in enumerate(conversation_history[:-1]) if msg["role"] == "user"
+    ]
+    
+    return conversation_history, chat_display
 
 theme = gr.themes.Ocean(
     text_size="lg",
@@ -83,8 +85,6 @@ custom_css = """
 """
 
 def clear_history():
-    global conversation_history
-    conversation_history = []
     return []
 
 
@@ -106,8 +106,9 @@ with gr.Blocks(theme = theme, css = custom_css) as demo:
 
     custom_country_input = gr.Textbox(label="Enter Country (if not listed)", visible=False)
 
-    chatbot = gr.Chatbot(label="💬 Chat History")
-    chatbot.clear(fn=clear_history, outputs=chatbot)
+    chatbot = gr.Chatbot(label="💬 Chat History", type="messages")
+    chatbot.clear(fn=clear_history, outputs=conversation_state)
+    
     with gr.Row():
         family_btn = gr.Button("👨‍👩‍👧‍👦 Family", elem_classes="law-button")
         corporate_btn = gr.Button("🏢 Corporate", elem_classes="law-button")
@@ -138,17 +139,22 @@ with gr.Blocks(theme = theme, css = custom_css) as demo:
 
     submit_btn = gr.Button("Send",variant="primary")
 
-    def submit(country, custom_country, language, scenario, chat_history):
+    def submit(country, custom_country, language, scenario, conversation_state):
         selected_country = custom_country if country == "Other" else country
-        new_chat_history = get_chatbot_response(scenario, selected_country, language)
-        return new_chat_history, ""
+        updated_history, chat_display = get_chatbot_response(
+            scenario, selected_country, language, conversation_state
+        )
+        return updated_history, chat_display, ""
+
 
     country_input.change(lambda c: gr.update(visible=c == "Other"), inputs=country_input, outputs=custom_country_input)
 
+    conversation_state = gr.State([])
+
     submit_btn.click(
         submit, 
-        inputs=[country_input, custom_country_input, language_input, scenario_input, chatbot], 
-        outputs=[chatbot, scenario_input]
+        inputs=[country_input, custom_country_input, language_input, scenario_input, conversation_state], 
+        outputs=[conversation_state, chatbot, scenario_input]
     )
 
 demo.launch()