Update app.py

app.py

@@ -28,7 +28,7 @@ def generate(prompt, history, temperature=0.1, max_new_tokens=256, top_p=0.95, r
     if temperature < 1e-2:
         temperature = 1e-2
     top_p = float(top_p)
-    
+
     generate_kwargs = dict(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
@@ -37,20 +37,25 @@ def generate(prompt, history, temperature=0.1, max_new_tokens=256, top_p=0.95, r
         do_sample=True,
         seed=42,
     )
-    
+
     # Translate the Arabic prompt to English
     english_prompt = translate_to_english(prompt)
-    
+
     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 before yielding it
-        arabic_output = translate_to_arabic(output)
-        yield arabic_output
-    return arabic_output
+
+    # Translate the English response back to Arabic
+    arabic_output = translate_to_arabic(output)
+
+    # Update the history state
+    history.append((prompt, arabic_output))
+
+    # Return the response and the updated state
+    return arabic_output, history
 
 # Additional input widgets for controlling the generation parameters
 additional_inputs = [
@@ -93,14 +98,19 @@ additional_inputs = [
 ]
 
 # Creating and launching the Gradio interface
-gr.Interface(
+iface = gr.Interface(
     fn=generate,
     inputs=[
-        gr.Textbox(lines=2, label="Your Prompt in Arabic"),
-        gr.State(),
-        *additional_inputs
+        gr.Textbox(lines=2, placeholder="Enter your prompt in Arabic"),
+        gr.State()  # State input to maintain the conversation history
+    ] + additional_inputs,
+    outputs=[
+        gr.Textbox(placeholder="Generated response in Arabic"),
+        gr.State()  # State output to maintain the conversation history
     ],
-    outputs=gr.Textbox(label="Generated Response in Arabic"),
     title="Try Arabic Misteral",
     description="Interact with an advanced AI model in Arabic. Adjust the settings below to tailor the responses. Your prompts will be translated to English, processed by the AI, and the response will be translated back to Arabic."
-).launch(show_api=True)
+)
+
+# Launch the interface
+iface.launch()