Update app.py

app.py

@@ -1,12 +1,13 @@
 import streamlit as st
 import requests
 import os
+import time  # Import time module for implementing delay
 
 # Fetch Hugging Face and Groq API keys from secrets
-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')
+Transalate_token = os.getenv('Translate')
+Image_Token = os.getenv('Image_generation')
+Content_Token = os.getenv('ContentGeneration')
+Image_prompt_token = os.getenv('Prompt_generation')
 
 # API Headers
 Translate = {"Authorization": f"Bearer {Transalate_token}"}
@@ -44,18 +45,25 @@ content_models = {
 # Default content generation model
 default_content_model = "llama-3.1-70b-versatile"
 
-# Function to query Hugging Face translation model
+# Function to query Hugging Face translation model with retry mechanism
 def translate_text(text):
     payload = {"inputs": text}
-    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(f'Please try after sometime 😥😥😥')
-        return None
+    max_retries = 3  # Maximum number of retry attempts
+    delay = 2  # Delay between retries in seconds
+
+    for attempt in range(max_retries):
+        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.warning(f"Translation failed (Attempt {attempt+1}/{max_retries}) - Retrying in {delay} seconds...")
+            time.sleep(delay)  # Wait for 2 seconds before retrying
+
+    # If all retries fail, show an error
+    st.error(f"Translation failed after {max_retries} attempts. Please reload the page and try again later.")
+    return None
 
 # Function to query Groq content generation model
 def generate_content(english_text, max_tokens, temperature, model):
@@ -105,8 +113,26 @@ def generate_image(image_prompt, model_url):
         st.error(f"Image Generation Error {response.status_code}: {response.text}")
         return None
 
+# User Guide Section
+def show_user_guide():
+    st.title("FusionMind User Guide")
+    st.write("""
+    ### Welcome to the FusionMind User Guide!
+
+    ### How to use this app:
+    ... (omitted for brevity)
+    """)
+
 # Main Streamlit app
 def main():
+    # Sidebar Menu
+    st.sidebar.title("FusionMind Options")
+    page = st.sidebar.radio("Select a page:", ["Main App", "User Guide"])
+
+    if page == "User Guide":
+        show_user_guide()
+        return
+
     # Custom CSS for background, borders, and other styling
     st.markdown(
         """
@@ -153,7 +179,7 @@ def main():
     st.sidebar.header("Settings")
     temperature = st.sidebar.slider("Select Temperature", 0.1, 1.0, 0.7)
     max_tokens = st.sidebar.slider("Max Tokens for Content Generation", 100, 400, 200)
-    
+
     # Content generation model selection
     content_model = st.sidebar.selectbox("Select Content Generation Model", list(content_models.keys()), index=0)
 
@@ -161,11 +187,11 @@ def main():
     image_model = st.sidebar.selectbox("Select Image Generation Model", list(image_generation_urls.keys()), index=0)
 
     # Reminder about model availability
-    st.sidebar.warning("Note: Based on availability, some models might not work. Please try another model if an error occurs.")
+    st.sidebar.warning("Note: Based on availability, some models might not work. Please try another model if an error occurs.By default the perfect model is selected try with it and then experiment with different models")
 
     # Suggested inputs
     st.write("## Suggested Inputs")
-    suggestions = ["தரவு அறிவியல்", "புதிய திறன்களைக் கற்றுக்கொள்வது எப்படி", "ராக்கெட் எப்படி வேலை செய்கிறது"]
+    suggestions = ["தரவு அறிவியல்", "உளவியல்", "ராக்கெட் எப்படி வேலை செய்கிறது"]
     selected_suggestion = st.selectbox("Select a suggestion or enter your own:", [""] + suggestions)
 
     # Input box for user
@@ -177,22 +203,27 @@ def main():
             st.write("### Translated English Text:")
             english_text = translate_text(tamil_input)
             if english_text:
-                st.success(english_text)
-
-                # Step 2: Generate Educational Content
-                st.write("### Generated Educational Content:")
-                with st.spinner('Generating content...'):
-                    content_output = generate_content(english_text, max_tokens, temperature, content_models[content_model])
-                    if content_output:
-                        st.success(content_output)
-
-                # Step 3: Generate Image from the prompt
-                st.write("### Generated Image:")
-                with st.spinner('Generating image...'):
+                st.write(english_text)
+
+                # Step 2: Content Generation
+                st.write("### Educational Content Generated:")
+                content = generate_content(english_text, max_tokens, temperature, content_models[content_model])
+                if content:
+                    st.write(content)
+
+                    # Step 3: Generate Image Prompt
+                    st.write("### Image Prompt:")
                     image_prompt = generate_image_prompt(english_text)
-                    image_data = generate_image(image_prompt, image_generation_urls[image_model])
-                    if image_data:
-                        st.image(image_data, caption="Generated Image")
+                    if image_prompt:
+                        st.write(image_prompt)
+
+                        # Step 4: Image Generation
+                        st.write("### Generated Image:")
+                        image = generate_image(image_prompt, image_generation_urls[image_model])
+                        if image:
+                            st.image(image)
+        else:
+            st.error("Please enter or select Tamil text.")
 
 if __name__ == "__main__":
     main()