Update app.py

app.py

@@ -1,40 +1,41 @@
 import os
-import openai
+import requests
+import json
 import gradio as gr
 
-# Set up OpenAI API credentials
-api_key = os.environ.get("OPENAI_API_KEY")
-openai.api_key = api_key
+# Set up Hugging Face API credentials
+api_key = os.environ.get("HUGGINGFACE_API_KEY")
 
 # Define a function to generate a response to user input
 def generate_response(user_input):
-    # Check if user input contains the phrase "who created you"
-    if "who created you" in user_input.lower():
-        # Generate a response that includes your name
-        chat_response = "I was created by RamV!"
-    else:
-        # Construct the prompt with the user input
-        prompt = f"You said: {user_input}"
-
-        # Generate a response using the OpenAI API
-        response = openai.Completion.create(
-            model="text-davinci-003",
-            prompt=prompt,
-            temperature=0.9,
-            max_tokens=150,
-            top_p=1,
-            frequency_penalty=0,
-            presence_penalty=0.6,
-            stop=["Human:", "AI:"]
-        )
-
-        # Extract the response from the API output
-        chat_response = response.choices[0].text.strip()
+    # Construct the payload for the API request
+    payload = {
+        "inputs": user_input,
+        "parameters": {
+            "max_new_tokens": 100,
+            "temperature": 0.5,
+            "model": "microsoft/DialoGPT-medium"
+        }
+    }
+
+    # 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
+    response = requests.post(
+        "https://api-inference.huggingface.co/models/openai/dialogue",
+        headers=headers,
+        data=json.dumps(payload)
+    )
+    chat_response = response.json()["generated_text"].strip()
 
     return chat_response
 
 # Define the function to handle the chat history
-def openai_chat_history(input, history):
+def huggingface_chat_history(input, history):
     history = history or []
     if input.strip() != "":
         s = list(sum(history, ()))
@@ -51,11 +52,9 @@ conversation_prompt = "Welcome to ChatRobo, kindly type in your enquiries: "
 
 # Set up the Gradio interface
 block = gr.Interface(
-    fn=openai_chat_history, 
+    fn=huggingface_chat_history, 
     inputs=[gr.inputs.Textbox(placeholder=conversation_prompt)], 
     outputs=[gr.outputs.Textbox(label="ChatRobo Output")]
 )
-
-# Launch the Gradio interface
-block.launch()
+block.run()