Update app.py

app.py

@@ -9,8 +9,8 @@ load_dotenv()
 image_API="https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
 translation_API="https://api-inference.huggingface.co/models/facebook/nllb-200-distilled-1.3B"
 whisper_API="https://api-inference.huggingface.co/models/openai/whisper-large-v3"
-txt_API="https://api-inference.huggingface.co/models/openai-community/gpt2"
-HF_TOKEN=os.getenv("HF_Token")
+txt_API="https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-R1"
+HF_TOKEN=os.getenv("HF_TOKEN")
 login(HF_TOKEN)
 headers={"Authorization":f"Bearer {HF_TOKEN}"}
 
@@ -49,8 +49,21 @@ def query_image(prompt):
     time.sleep(delay)
   return None
 
-  
-
+def query_text_generation(prompt):
+  max_retries=5
+  delay=10
+  payload={"inputs":f"give me a short story about {prompt}"}
+  for attempt in range(max_retries):
+    response=requests.post(txt_API,headers=headers,json=payload)
+    if response.status_code == 200:
+      result=response.json()
+      if isinstance(result,list) and len(result)>0:
+        return result[0].get("generated_text","Text Generation Error")
+      elif isinstance(result,dict) and "generated_text" in result:
+        return result["generated_text"]
+    print(f"⚠️ Text generation failed, retrying in {delay} seconds...")
+    time.sleep(delay)
+  return None
 
 def process_audio(audio_path):
   if not audio_path:
@@ -61,6 +74,7 @@ def process_audio(audio_path):
     translation=query_translation(tamil_text)
     translated_text=translation.get("translated_text","Translation error")
     image_path=query_image(translated_text)
+    story=query_text_generation(translated_text)
     return tamil_text,translated_text,image_path
   except Exception as e:
     return None,str(e),None