Update app.py

app.py

@@ -1,61 +1,35 @@
-import os
-import requests
-import json
 import gradio as gr
+import requests
 
-# Set up Hugging Face API credentials
-api_key = os.environ.get("OPENAI_API_KEY")
-
-# Define a function to generate a response to user input
-def generate_response(user_input):
-    # Construct the payload for the API request
+def huggingface_chat_history(input, history=[]):
+    # create payload dictionary
     payload = {
-        "inputs": user_input,
+        "inputs": input,
         "parameters": {
-            "max_new_tokens": 100,
-            "temperature": 0.5,
-            "model": "microsoft/DialoGPT-medium"
+            "history": history
         }
     }
 
-    # Set up the request headers with the API key
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-
-    # Make the API request and extract the response
+    # make request to Hugging Face model API
     response = requests.post(
-        "https://api-inference.huggingface.co/models/openai/dialogue",
-        headers=headers,
-        data=json.dumps(payload)
+        "https://api-inference.huggingface.co/models/microsoft/DialoGPT-medium", 
+        headers={"Authorization": "OPENAI_API_KEY"}, 
+        json=payload
     )
-    chat_response = response.json()["generated_text"].strip()
 
+    # extract chatbot response from API response
+    chat_response = response.json()[0]['generated_text']
+    history.append([input, chat_response])
     return chat_response
 
-# Define the function to handle the chat history
-def huggingface_chat_history(input, history):
-    history = history or []
-    if input.strip() != "":
-        s = list(sum(history, ()))
-        s.append(input)
-        inp = ' '.join(s)
-        output = generate_response(inp)
-        history.append((input, output))
-        return history[-1][1]
-    else:
-        return ""
-
-# Define the conversation prompt
 conversation_prompt = "Welcome to ChatRobo, kindly type in your enquiries: "
 
-# Set up the Gradio interface
-block = gr.Interface(
-    fn=huggingface_chat_history, 
-    inputs=[gr.inputs.Textbox(placeholder=conversation_prompt)], 
-    outputs=[gr.outputs.Textbox(label="ChatRobo Output")]
+iface = gr.Interface(
+    fn=huggingface_chat_history,
+    inputs=gr.inputs.Textbox(prompt=conversation_prompt),
+    outputs=gr.outputs.Textbox()
 )
-dir(gr.Interface)
+
+iface.launch()