Update app.py

app.py

@@ -3,13 +3,20 @@ import requests
 import os
 
 # Fetch Hugging Face and Groq API keys from secrets
-HUGGINGFACE_TOKEN = os.getenv('HUGGINGFACE_TOKEN')
-GROQ_API_KEY = os.getenv('GROQ_API_KEY')
+Transalate_token = os.getenv('HUGGINGFACE_TOKEN')
+Image_Token = os.getenv('HUGGINGFACE_TOKEN')
+Content_Token = os.getenv('GROQ_API_KEY')
+Image_prompt_token = os.getenv('GROQ_API_KEY')
 
 # API Headers
-headers_hf = {"Authorization": f"Bearer {HUGGINGFACE_TOKEN}"}
-headers_groq = {
-    "Authorization": f"Bearer {GROQ_API_KEY}",
+Translate = {"Authorization": f"Bearer {Transalate_token}"}
+Image_generation = {"Authorization": f"Bearer {Image_Token}"}
+Content_generation = {
+    "Authorization": f"Bearer {Content_Token}",
+    "Content-Type": "application/json"
+}
+Image_Prompt = {
+    "Authorization": f"Bearer {Image_prompt_token}",
     "Content-Type": "application/json"
 }
 
@@ -22,13 +29,14 @@ image_generation_url = "https://api-inference.huggingface.co/models/black-forest
 # Function to query Hugging Face translation model
 def translate_text(text):
     payload = {"inputs": text}
-    response = requests.post(translation_url, headers=headers_hf, json=payload)
+    response = requests.post(translation_url, headers=Translate, json=payload)
     if response.status_code == 200:
         result = response.json()
         translated_text = result[0]['generated_text']
         return translated_text
     else:
         st.error(f"Translation Error {response.status_code}: {response.text}")
+        st.write('Please try again or provide an English input 😥')
         return None
 
 # Function to query Groq content generation model
@@ -43,7 +51,7 @@ def generate_content(english_text, max_tokens, temperature):
         "max_tokens": max_tokens,
         "temperature": temperature
     }
-    response = requests.post(url, json=payload, headers=headers_groq)
+    response = requests.post(url, json=payload, headers=Content_generation)
     if response.status_code == 200:
         result = response.json()
         return result['choices'][0]['message']['content']
@@ -61,7 +69,7 @@ def generate_image_prompt(english_text):
         ],
         "max_tokens": 30
     }
-    response = requests.post("https://api.groq.com/openai/v1/chat/completions", json=payload, headers=headers_groq)
+    response = requests.post("https://api.groq.com/openai/v1/chat/completions", json=payload, headers=Image_Prompt)
     if response.status_code == 200:
         result = response.json()
         return result['choices'][0]['message']['content']
@@ -72,7 +80,7 @@ def generate_image_prompt(english_text):
 # Function to generate an image from the prompt
 def generate_image(image_prompt):
     data = {"inputs": image_prompt}
-    response = requests.post(image_generation_url, headers=headers_hf, json=data)
+    response = requests.post(image_generation_url, headers=Image_generation, json=data)
     if response.status_code == 200:
         return response.content
     else:
@@ -81,8 +89,37 @@ def generate_image(image_prompt):
 
 # Main Streamlit app
 def main():
-    st.title("Multimodal Generator")
-
+    # Custom CSS for background, borders, and other styling
+    st.markdown(
+        """
+        <style>
+        body {
+            background-image: url('https://wallpapercave.com/wp/wp4008910.jpg');
+            background-size: cover;
+        }
+        .reportview-container {
+            background: rgba(255, 255, 255, 0.85);
+            padding: 2rem;
+            border-radius: 10px;
+            box-shadow: 0px 0px 20px rgba(0, 0, 0, 0.1);
+        }
+        .result-container {
+            border: 2px solid #4CAF50;
+            padding: 20px;
+            border-radius: 10px;
+            margin-top: 20px;
+            animation: fadeIn 2s ease;
+        }
+        @keyframes fadeIn {
+            0% { opacity: 0; }
+            100% { opacity: 1; }
+        }
+        </style>
+        """, unsafe_allow_html=True
+    )
+
+    st.title("🌐 Multimodal Generator")
+    
     # Sidebar for temperature and token adjustment
     st.sidebar.header("Settings")
     temperature = st.sidebar.slider("Select Temperature", 0.1, 1.0, 0.7)
@@ -101,6 +138,9 @@ def main():
         if tamil_input:
             st.write("### Translated English Text:")
             english_text = translate_text(tamil_input)
+            if not english_text:
+                english_text = st.text_input("Translation failed. Please enter English text instead:")
+        
             if english_text:
                 st.success(english_text)
 
@@ -121,4 +161,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-