Update app.py

app.py

@@ -27,7 +27,10 @@ MODEL_OPTIONS = [
 
 def generate_text(input_text, selected_model, history):
     if history is None:
-        history = ""  # Initialize history if it's None
+        history = []  # Initialize history as a list if it's None
+
+    # Append the user's input to the conversation history
+    history.append({"role": "user", "content": input_text})
     
     # Call the API to get the model response
     response = requests.post(
@@ -37,8 +40,8 @@ def generate_text(input_text, selected_model, history):
             "Content-Type": "application/json"
         },
         data=json.dumps({
-            "model": selected_model,  # Use selected model
-            "messages": [{"role": "user", "content": input_text}],
+            "model": selected_model,  # Use the selected model
+            "messages": history,  # Send the entire conversation history
             "top_p": 1,
             "temperature": 1,
             "frequency_penalty": 0,
@@ -59,10 +62,14 @@ def generate_text(input_text, selected_model, history):
     except json.JSONDecodeError:
         generated_response = "Error: Unable to parse response."
 
-    # Append the new conversation to the history
-    history += f"User: {input_text}\nModel: {selected_model}\nResponse: {generated_response}\n\n"
+    # Append the model's response to the conversation history
+    history.append({"role": "assistant", "content": generated_response})
+    
+    # Format the history for display
+    display_history = "\n\n".join([f"{msg['role'].capitalize()}: {msg['content']}" for msg in history])
     
-    return generated_response, history, history  # Return three values
+    return generated_response, display_history, history  # Return the updated history for continuity
+
 
 # Inject custom CSS directly into the Gradio interface for scrollbars
 css = """