Update app.py

app.py

@@ -35,14 +35,14 @@ def translate_text(tamil_text):
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
-# API credentials and endpoint
-API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
-headers = {"Authorization": f"Bearer {hf_token}"}
+# API credentials and endpoint for FLUX
+flux_API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
+flux_headers = {"Authorization": f"Bearer {hf_token}"}
 
 # Function to send payload and generate image
 def generate_image(prompt):
     try:
-        response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
+        response = requests.post(flux_API_URL, headers=flux_headers, json={"inputs": prompt})
 
         # Check if the response is successful
         if response.status_code == 200:
@@ -63,19 +63,23 @@ def generate_image(prompt):
         print(f"An error occurred: {e}")
         return None
 
-# Import necessary libraries for Mistral model
-from transformers import AutoTokenizer, AutoModelForCausalLM
+# Function for Mistral API call to generate creative text
+mistral_API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-v0.1"
+mistral_headers = {"Authorization": f"Bearer {hf_token}"}
 
-# Load Mistral model and tokenizer
-mistral_tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
-mistral_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
-
-# Function to generate creative text based on translated text using Mistral
 def generate_creative_text(translated_text):
-    input_ids = mistral_tokenizer(translated_text, return_tensors='pt').input_ids
-    generated_text_ids = mistral_model.generate(input_ids, max_length=100)
-    creative_text = mistral_tokenizer.decode(generated_text_ids[0], skip_special_tokens=True)
-    return creative_text
+    try:
+        response = requests.post(mistral_API_URL, headers=mistral_headers, json={"inputs": translated_text})
+        if response.status_code == 200:
+            creative_text = response.json()[0]['generated_text']
+            return creative_text
+        else:
+            print(f"Failed to get creative text: Status code {response.status_code}")
+            print("Response content:", response.text)  # Print response for debugging
+            return "Error generating creative text"
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return None
 
 # Function to handle the full workflow
 def translate_generate_image_and_text(tamil_text):