Update app.py

app.py

@@ -1,26 +1,25 @@
 import gradio as gr
 import requests
+import os
 
-# Simulating the API Key as "legal"
-API_KEY = 'sk-7ac906276b784f45a9a989aafccef946'
 
-# Chatbot Function using DeepSeek LLM
+
+# Hugging Face Inference API Token
+API_URL = "https://api-inference.huggingface.co/models/gpt2"
+API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")  # Replace with your Hugging Face API token
+
+# Chatbot Function using Hugging Face Inference API
 def ielts_chat(user_input, part):
     prompt = f"You are an IELTS tutor. Answer the following question in detail as if you were helping a student prepare for the IELTS Speaking test.\n\nPart: {part}\nQuestion: {user_input}\n\nAnswer:"
 
     response = requests.post(
-        "https://api.deepseek.com/v1/completions",
-        headers={"Authorization": f"Bearer {API_KEY}"},
-        json={
-            "model": "deepseek-llm",
-            "prompt": prompt,
-            "max_tokens": 150,
-            "temperature": 0.7
-        }
+        API_URL,
+        headers={"Authorization": f"Bearer {API_TOKEN}"},
+        json={"inputs": prompt}
     )
 
     if response.status_code == 200:
-        return response.json()['choices'][0]['text'].strip()
+        return response.json()[0]['generated_text'].strip()
     else:
         return f"Error {response.status_code}: {response.text}"
 
@@ -33,8 +32,7 @@ iface = gr.Interface(
     ],
     outputs="text",
     title="IELTS Speaking Practice Bot",
-    description="Ask a question related to IELTS Speaking, and the bot will respond as if it were an IELTS tutor using AI-generated responses from DeepSeek LLM."
+    description="Ask a question related to IELTS Speaking, and the bot will respond as if it were an IELTS tutor using AI-generated responses from Hugging Face's GPT-2 model."
 )
 
 iface.launch()
-