Update app.py

app.py

@@ -23,7 +23,11 @@ def format_prompt(message, history):
     return prompt
 
 # The main function to generate responses
-def generate(prompt, history, temperature=0.1, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
+def generate(prompt, history=None, temperature=0.1, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
+    # Initialize history as an empty list if it's None
+    if history is None:
+        history = []
+
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -41,17 +45,19 @@ def generate(prompt, history, temperature=0.1, max_new_tokens=256, top_p=0.95, r
     # Translate the Arabic prompt to English
     english_prompt = translate_to_english(prompt)
 
+    # Format the prompt with the conversation history
     formatted_prompt = format_prompt(english_prompt, history)
+    
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
     
     for response in stream:
         output += response.token.text
-
+    
     # Translate the English response back to Arabic
     arabic_output = translate_to_arabic(output)
 
-    # Update the history state
+    # Update the history state with the latest exchange
     history.append((prompt, arabic_output))
 
     # Return the response and the updated state