Update app.py

app.py

@@ -11,7 +11,7 @@ from streamlit_lottie import st_lottie
 from streamlit_option_menu import option_menu
 import time
 
-# Set page config
+# Set page configuration
 st.set_page_config(
     page_title="TeleGuide | AI Telecom Assistant",
     page_icon="🛰️",
@@ -19,7 +19,7 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Load Lottie animation
+# Function to load Lottie animations
 def load_lottie(url: str):
     try:
         r = requests.get(url)
@@ -29,23 +29,18 @@ def load_lottie(url: str):
     except:
         return None
 
-# Custom CSS with animations and improved styling
+# Apply custom CSS for styling and animations
 st.markdown("""
 <style>
-    /* Main app styling */
     .stApp {
         background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
     }
-    
-    /* Card-like containers */
     .css-1r6slb0 {
         background: white;
         border-radius: 20px;
         padding: 20px;
         box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
     }
-    
-    /* Header styling */
     .main-header {
         font-size: 2.5rem;
         font-weight: 700;
@@ -54,8 +49,6 @@ st.markdown("""
         margin-bottom: 2rem;
         animation: fadeIn 1.5s ease-in;
     }
-    
-    /* Button styling */
     .stButton>button {
         width: 100%;
         border-radius: 10px;
@@ -65,13 +58,10 @@ st.markdown("""
         padding: 0.5rem 1rem;
         transition: all 0.3s ease;
     }
-    
     .stButton>button:hover {
         transform: translateY(-2px);
         box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2);
     }
-    
-    /* Input field styling */
     .stTextInput>div>div>input,
     .stTextArea>div>div>textarea {
         border-radius: 10px;
@@ -79,23 +69,16 @@ st.markdown("""
         padding: 10px;
         background-color: white;
     }
-    
-    /* Sidebar styling */
     .css-1d391kg {
         background: linear-gradient(180deg, #1E3D59 0%, #2193b0 100%);
     }
-    
-    /* Animations */
     @keyframes fadeIn {
         from { opacity: 0; transform: translateY(20px); }
         to { opacity: 1; transform: translateY(0); }
     }
-    
     .fade-in {
         animation: fadeIn 1s ease-in;
     }
-    
-    /* Status messages */
     .success-message {
         padding: 1rem;
         border-radius: 10px;
@@ -103,7 +86,6 @@ st.markdown("""
         color: #155724;
         margin: 1rem 0;
     }
-    
     .error-message {
         padding: 1rem;
         border-radius: 10px;
@@ -114,11 +96,13 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
-# Initialize the OpenAI client with error handling
+# Initialize OpenAI client with Hugging Face secrets for API key
 @st.cache_resource
 def get_openai_client():
     try:
-        return OpenAI(api_key=st.secrets["api_key"], base_url="https://api.together.xyz")
+        # Fetch the API key from Hugging Face secrets
+        api_key = st.secrets["api_key"]
+        return OpenAI(api_key=api_key, base_url="https://api.together.xyz")
     except Exception as e:
         st.error(f"Error initializing API client: {str(e)}")
         return None
@@ -129,6 +113,7 @@ client = get_openai_client()
 lottie_telecom = load_lottie("https://assets4.lottiefiles.com/packages/lf20_qz3tpn4w.json")
 lottie_analysis = load_lottie("https://assets4.lottiefiles.com/packages/lf20_xh83pj1k.json")
 
+# Process text query function
 def process_text_query(query, model="meta-llama/Llama-3.2-3B-Instruct-Turbo"):
     try:
         with st.spinner("Processing your query..."):
@@ -146,6 +131,7 @@ def process_text_query(query, model="meta-llama/Llama-3.2-3B-Instruct-Turbo"):
         st.error(f"Error processing query: {str(e)}")
         return None
 
+# Process image query function
 def process_image_query(image_base64, query, model="meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo"):
     try:
         with st.spinner("Analyzing image..."):
@@ -170,6 +156,7 @@ def process_image_query(image_base64, query, model="meta-llama/Llama-3.2-90B-Vis
         st.error(f"Error analyzing image: {str(e)}")
         return None
 
+# Convert image to base64
 def image_to_base64(image):
     try:
         buffered = io.BytesIO()
@@ -179,7 +166,7 @@ def image_to_base64(image):
         st.error(f"Error converting image: {str(e)}")
         return None
 
-# Sidebar
+# Sidebar content
 with st.sidebar:
     st.title("🛰️ TeleGuide")
     st_lottie(lottie_telecom, height=200)
@@ -199,7 +186,7 @@ with st.sidebar:
 # Main content
 st.markdown('<h1 class="main-header">Welcome to TeleGuide</h1>', unsafe_allow_html=True)
 
-# Navigation
+# Navigation menu
 selected = option_menu(
     menu_title=None,
     options=["Text Analysis", "Document Processing", "Image Analysis"],
@@ -220,6 +207,7 @@ selected = option_menu(
     }
 )
 
+# Handle different options from the navigation menu
 if selected == "Text Analysis":
     st.markdown("### 💬 Text Analysis")
     st_lottie(lottie_analysis, height=200)
@@ -245,51 +233,36 @@ elif selected == "Document Processing":
     text_input = st.text_area("Enter document text:", height=150)
     if st.button("Analyze Document"):
         if text_input:
-            analysis_prompt = f"Analyze the following {document_type.lower()}: {text_input}"
-            response = process_text_query(analysis_prompt)
+            response = process_text_query(f"Analyze the following {document_type}: {text_input}")
             if response:
                 st.markdown('<div class="success-message">✅ Document analyzed successfully!</div>', unsafe_allow_html=True)
-                st.markdown("### Analysis Results:")
+                st.markdown("### Response:")
                 st.write(response)
         else:
-            st.warning("Please enter document text.")
+            st.warning("Please enter some text to analyze.")
 
 elif selected == "Image Analysis":
     st.markdown("### 🖼️ Image Analysis")
-    image_type = st.selectbox(
-        "Select Image Type",
-        ["Network Infrastructure", "Equipment", "Site Survey", "Signal Coverage"]
-    )
+    image_file = st.file_uploader("Upload an image for analysis", type=["jpg", "jpeg", "png"])
     
-    uploaded_file = st.file_uploader("Upload image...", type=["jpg", "png", "jpeg"])
-    if uploaded_file:
-        try:
-            image = Image.open(uploaded_file)
-            st.image(image, caption="Uploaded Image", use_column_width=True)
-            query = st.text_area("Analysis requirements:", height=100)
-            
-            if st.button("Analyze Image"):
-                if query:
-                    image_base64 = image_to_base64(image)
-                    if image_base64:
-                        response = process_image_query(image_base64, query)
-                        if response:
-                            st.markdown('<div class="success-message">✅ Image analyzed successfully!</div>', unsafe_allow_html=True)
-                            st.markdown("### Analysis Results:")
-                            st.write(response)
-                else:
-                    st.warning("Please enter analysis requirements.")
-        except Exception as e:
-            st.error(f"Error processing image: {str(e)}")
+    if image_file:
+        image = Image.open(image_file)
+        st.image(image, caption="Uploaded Image", use_column_width=True)
+        
+        query = st.text_input("Enter your query about this image:")
+        
+        if st.button("Analyze Image"):
+            image_base64 = image_to_base64(image)
+            if image_base64 and query:
+                response = process_image_query(image_base64, query)
+                if response:
+                    st.markdown('<div class="success-message">✅ Image analyzed successfully!</div>', unsafe_allow_html=True)
+                    st.markdown("### Response:")
+                    st.write(response)
+            else:
+                st.warning("Please upload an image and enter a query.")
 
 # Footer
 st.markdown("---")
-st.markdown(
-    """
-    <div style='text-align: center; color: #666;'>
-        <p>TeleGuide - Your AI-Powered Telecommunications Assistant</p>
-        <p>Made with ❤️ for telecom professionals</p>
-    </div>
-    """,
-    unsafe_allow_html=True
-)
+st.caption("🚀 Powered by OpenAI | Streamlit | Llama Models")
+