Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import streamlit as st
 import requests
 import json
@@ -6,10 +7,13 @@ import matplotlib.pyplot as plt
 import numpy as np
 from io import BytesIO
 
-# Hugging Face API URLs and token
+# Load tokens from environment variables
+HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
+GROQ_TOKEN = os.getenv("GROQ_TOKEN")
+
+# Hugging Face API URLs
 TRANSLATE_API_URL = "https://api-inference.huggingface.co/models/facebook/mbart-large-50-many-to-one-mmt"
 IMAGE_API_URL = "https://api-inference.huggingface.co/models/CompVis/stable-diffusion-v1-4"
-HUGGINGFACE_TOKEN = "YOUR_HUGGINGFACE_TOKEN"  # Replace with your actual token
 headers = {"Authorization": f"Bearer {HUGGINGFACE_TOKEN}"}
 
 # Function to translate Tamil to English
@@ -41,7 +45,9 @@ def generate_text(prompt, max_tokens, temperature):
         "max_tokens": max_tokens,
         "temperature": temperature,
     }
-    response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers={"Authorization": "Bearer YOUR_GROQ_TOKEN"}, json=payload)
+    response = requests.post("https://api.groq.com/openai/v1/chat/completions", 
+                              headers={"Authorization": f"Bearer {GROQ_TOKEN}"}, 
+                              json=payload)
     if response.status_code == 200:
         result = response.json()
         return result["choices"][0]["message"]["content"]
@@ -82,10 +88,10 @@ if st.sidebar.button("Submit"):
 
 # Guide section
 if st.sidebar.button("How to Use"):
-    st.sidebar.write("""
-    ### How to Use This App
-    1. **Enter Tamil Text**: Type or paste the Tamil text you want to translate.
-    2. **Adjust Settings**: Use the sliders to set the maximum tokens for text generation and the temperature for creativity.
-    3. **Submit**: Click the Submit button to generate translations, images, and stories.
-    4. **View Results**: Check the translated text, generated image, and story in the main area.
+    st.sidebar.write(""" 
+    ### How to Use This App 
+    1. **Enter Tamil Text**: Type or paste the Tamil text you want to translate. 
+    2. **Adjust Settings**: Use the sliders to set the maximum tokens for text generation and the temperature for creativity. 
+    3. **Submit**: Click the Submit button to generate translations, images, and stories. 
+    4. **View Results**: Check the translated text, generated image, and story in the main area. 
     """)