Update app.py

app.py

@@ -21,7 +21,7 @@ MODEL_OPTIONS = [
     "liquid/lfm-40b:free"
 ]
 
-def generate_text(input_text, selected_model):
+def generate_text(input_text, selected_model, history):
     response = requests.post(
         url="https://openrouter.ai/api/v1/chat/completions",
         headers={
@@ -42,14 +42,19 @@ def generate_text(input_text, selected_model):
     
     # Handle errors
     if response.status_code != 200:
-        return f"Error: {response.status_code}, {response.text}"
+        return f"Error: {response.status_code}, {response.text}", history
     
     # Parse and return the content of the response
     try:
         response_json = response.json()
-        return response_json.get("choices", [{}])[0].get("message", {}).get("content", "No content returned.")
+        generated_response = response_json.get("choices", [{}])[0].get("message", {}).get("content", "No content returned.")
     except json.JSONDecodeError:
-        return "Error: Unable to parse response."
+        generated_response = "Error: Unable to parse response."
+
+    # Append the new interaction to the history
+    history += f"User: {input_text}\nResponse: {generated_response}\n"
+    
+    return generated_response, history
 
 # Define the Gradio layout using Blocks
 with gr.Blocks() as iface:
@@ -73,7 +78,7 @@ with gr.Blocks() as iface:
     
     # Trigger the function when the user inputs text
     generate_button = gr.Button("Generate")
-    generate_button.click(generate_text, inputs=[input_text, selected_model], outputs=[output_response, output_history])
+    generate_button.click(generate_text, inputs=[input_text, selected_model, gr.State()], outputs=[output_response, output_history])
 
 # Launch the interface
 iface.launch()