Update app.py

app.py

@@ -10,47 +10,31 @@ def query_huggingface(payload):
     return response.json()
 
 # Function to generate response
-def generate_response(user_message, system_message="You are a helpful assistant.", history=[]):
-    messages = [{"role": "system", "content": system_message}]
-
-    for user, assistant in history:
-        if user:
-            messages.append({"role": "user", "content": user})
-        if assistant:
-            messages.append({"role": "assistant", "content": assistant})
-
-    messages.append({"role": "user", "content": user_message})
-
+def generate_response(user_message):
     payload = {
-        "inputs": messages,
-        "parameters": {
-            "max_tokens": 50,
-            "temperature": 0.7,
-            "top_p": 0.9,
-            "stream": False
-        }
+        "inputs": user_message
     }
 
     response = query_huggingface(payload)
-    output = response.get('generated_text', 'No response')
+    output = response.get('generated_text', 'No response')  # Adjust according to the model's response format
 
     return output
 
 # Gradio interface
-def gradio_interface(user_message, system_message, history):
-    response = generate_response(user_message, system_message, history)
+def gradio_interface(user_message, history):
+    response = generate_response(user_message)
     history.append((user_message, response))
     return history, history
 
 # Define Gradio blocks
 with gr.Blocks() as demo:
-    system_message = gr.Textbox(label="System Message", value="You are a helpful assistant.")
     user_message = gr.Textbox(label="Your Message")
     chatbot = gr.Chatbot(label="Chat History")
     state = gr.State([])
     submit_button = gr.Button("Submit")
 
-    submit_button.click(gradio_interface, inputs=[user_message, system_message, state], outputs=[chatbot, state])
+    submit_button.click(gradio_interface, inputs=[user_message, state], outputs=[chatbot, state])
 
 # Launch the interface
 demo.launch()
+