Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import requests
 import json
-import os 
+import os
 
 # Hugging Face API details
-API_URL = os.getenv("HF_TOKEN")
-HF_TOKEN = "YOUR_HUGGINGFACE_API_TOKEN"  # put your HF token here
+API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+HF_TOKEN = os.getenv("HF_TOKEN")  # make sure you set this in Hugging Face "Repository secrets"
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
 # Prompt template
@@ -33,12 +33,13 @@ Target language: {target_lang}
 Emotion: {emotion}
 """
 
-# Function to call Hugging Face API
 def query_hf(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+    try:
+        return response.json()
+    except:
+        return {"error": response.text}
 
-# Core translation function
 def translate(text, source_lang, target_lang, emotion):
     prompt = PROMPT_TEMPLATE.format(
         text=text,
@@ -49,35 +50,47 @@ def translate(text, source_lang, target_lang, emotion):
     payload = {"inputs": prompt, "parameters": {"temperature": 0.2}}
     output = query_hf(payload)
 
-    # Extract model output
+    # Debug check
+    if "error" in output:
+        return {
+            "cleaned_text": text,
+            "translated_text": "[Error: " + output["error"] + "]",
+            "emotion": emotion
+        }
+
     try:
         raw_text = output[0]["generated_text"]
         parsed = json.loads(raw_text.strip())
-    except Exception:
-        # If model messes up, fallback with safe JSON
+    except Exception as e:
         parsed = {
             "cleaned_text": text,
-            "translated_text": f"[Translation error]",
+            "translated_text": "[Parsing error: " + str(e) + "]",
             "emotion": emotion
         }
     return parsed
 
-# Gradio UI
-def gradio_interface(text, source_lang, target_lang, emotion):
+# Gradio UI with speech input
+def gradio_interface(audio, text, source_lang, target_lang, emotion):
+    # If user spoke, convert audio → text using Gradio’s built-in STT
+    if audio is not None:
+        text = audio  # Gradio’s Speech-to-Text returns text directly
+    if not text:
+        return {"error": "No input text provided"}
     result = translate(text, source_lang, target_lang, emotion)
     return json.dumps(result, indent=2, ensure_ascii=False)
 
 iface = gr.Interface(
     fn=gradio_interface,
     inputs=[
-        gr.Textbox(label="Input Text"),
+        gr.Audio(sources=["microphone"], type="text", label="🎙 Speech Input (or leave empty)"),
+        gr.Textbox(label="💬 Text Input"),
         gr.Radio(["en", "es"], label="Source Language"),
         gr.Radio(["en", "es"], label="Target Language"),
         gr.Radio(["happy", "sad", "angry", "calm", "excited"], label="Emotion")
     ],
     outputs=gr.Textbox(label="Output JSON"),
     title="AI Universal Translator - Translation Module",
-    description="Cleans text, translates EN ↔ ES, and preserves emotions."
+    description="Cleans text or speech, translates EN ↔ ES, and preserves emotions."
 )
 
 if __name__ == "__main__":