Update app.py

app.py

@@ -14,97 +14,179 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Custom CSS (keep your existing CSS here)
+# Custom CSS for futuristic design
 st.markdown("""
 <style>
-    /* Your existing CSS styles */
+    /* Main colors */
+    :root {
+        --primary: #6366f1;
+        --secondary: #10b981;
+        --dark: #1e293b;
+        --light: #f8fafc;
+    }
+    
+    /* Main container */
+    .stApp {
+        background: linear-gradient(135deg, #0f172a 0%, #1e293b 100%);
+        color: var(--light);
+    }
+    
+    /* Headers */
+    h1, h2, h3, h4, h5, h6 {
+        color: var(--light) !important;
+        font-family: 'Inter', sans-serif;
+    }
+    
+    /* Sidebar */
+    [data-testid="stSidebar"] {
+        background: linear-gradient(195deg, #0f172a 0%, #1e40af 100%) !important;
+    }
+    
+    /* Buttons */
+    .stButton>button {
+        background: var(--primary) !important;
+        color: white !important;
+        border: none;
+        border-radius: 8px;
+        padding: 10px 24px;
+        font-weight: 500;
+        transition: all 0.3s;
+    }
+    
+    .stButton>button:hover {
+        transform: translateY(-2px);
+        box-shadow: 0 4px 12px rgba(99, 102, 241, 0.3);
+    }
+    
+    /* File uploader */
+    [data-testid="stFileUploader"] {
+        border: 2px dashed var(--primary) !important;
+        border-radius: 12px !important;
+        padding: 20px !important;
+    }
+    
+    /* Markdown output */
+    .markdown-text {
+        background: rgba(30, 41, 59, 0.7) !important;
+        border-radius: 12px;
+        padding: 20px;
+        border-left: 4px solid var(--secondary);
+        animation: fadeIn 0.5s ease-in-out;
+    }
+    
+    @keyframes fadeIn {
+        from { opacity: 0; transform: translateY(10px); }
+        to { opacity: 1; transform: translateY(0); }
+    }
+    
+    /* Streamlit text input */
+    .stTextInput>div>div>input {
+        background: rgba(15, 23, 42, 0.7) !important;
+        color: white !important;
+        border: 1px solid #334155 !important;
+    }
 </style>
 """, unsafe_allow_html=True)
 
 # App title and description
 st.title("🔍 Optimus Alpha | Live Vision Assistant")
 
-# Initialize OpenAI client (keep your existing cached function)
+# Initialize OpenAI client
 @st.cache_resource
 def get_client():
     return OpenAI(
         base_url="https://openrouter.ai/api/v1",
-        api_key='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'
+        api_key='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'  # Replace with your actual key
     )
 
-# ===== New Live Camera Section =====
-st.subheader("Live Camera Feed")
-run_camera = st.checkbox("Enable Camera", value=False)
+# ===== Camera/Upload Selection =====
+input_method = st.radio(
+    "Select input method:",
+    ["Live Camera", "Upload Image"],
+    horizontal=True
+)
 
-FRAME_WINDOW = st.empty()
+# ===== Camera Section =====
 captured_image = None
 
-if run_camera:
-    cap = cv2.VideoCapture(0)
+if input_method == "Live Camera":
+    st.subheader("Live Camera Feed")
+    run_camera = st.checkbox("Start Camera", value=False)
     
-    capture_button = st.button("Capture Image")
-    stop_button = st.button("Stop Camera")
+    FRAME_WINDOW = st.empty()
     
-    if stop_button:
-        run_camera = False
-        cap.release()
-        st.experimental_rerun()
-    
-    while run_camera:
-        ret, frame = cap.read()
-        if not ret:
-            st.error("Failed to access camera")
-            break
-        
-        # Display the live feed
-        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        FRAME_WINDOW.image(frame)
-        
-        if capture_button:
-            captured_image = frame
+    if run_camera:
+        try:
+            cap = cv2.VideoCapture(0)
+            if not cap.isOpened():
+                st.error("Could not access camera. Please:")
+                st.markdown("""
+                - Check camera permissions
+                - Ensure no other app is using the camera
+                - Try reconnecting the camera
+                """)
+                run_camera = False
+            else:
+                capture_col, stop_col = st.columns(2)
+                with capture_col:
+                    capture_button = st.button("📸 Capture Image")
+                with stop_col:
+                    stop_button = st.button("🛑 Stop Camera")
+                
+                if stop_button:
+                    cap.release()
+                    st.rerun()
+                
+                while run_camera:
+                    ret, frame = cap.read()
+                    if not ret:
+                        st.error("Failed to capture frame")
+                        break
+                    
+                    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                    FRAME_WINDOW.image(frame)
+                    
+                    if capture_button:
+                        captured_image = frame
+                        cap.release()
+                        st.rerun()
+                        break
+        except Exception as e:
+            st.error(f"Camera error: {str(e)}")
             run_camera = False
-            cap.release()
-            break
-else:
-    FRAME_WINDOW.info("Camera is currently off")
-
-# ===== Image Processing Section =====
-col1, col2 = st.columns([1, 2])
 
-with col1:
-    st.subheader("Image Source")
-    
-    # Option to use captured image or upload
-    if captured_image is not None:
-        st.image(captured_image, caption="Captured Image", width=300)
-        use_captured = True
-    else:
-        use_captured = False
-    
+# ===== Upload Section =====
+else:
+    st.subheader("Upload Image")
     uploaded_file = st.file_uploader(
-        "Or upload an image",
+        "Choose an image file",
         type=["jpg", "jpeg", "png"],
-        disabled=use_captured
+        label_visibility="collapsed"
     )
+    if uploaded_file:
+        try:
+            captured_image = Image.open(uploaded_file)
+            st.image(captured_image, caption="Uploaded Image", width=300)
+        except Exception as e:
+            st.error(f"Error loading image: {str(e)}")
+
+# ===== Image Analysis Section =====
+if captured_image is not None:
+    st.subheader("AI Analysis")
     
-    # Determine which image to use
-    if use_captured:
+    # Convert to PIL Image if from OpenCV
+    if isinstance(captured_image, np.ndarray):
         image = Image.fromarray(captured_image)
-    elif uploaded_file:
-        image = Image.open(uploaded_file)
     else:
-        image = None
-
-with col2:
-    st.subheader("AI Analysis")
+        image = captured_image
     
     user_prompt = st.text_input(
-        "Your question about the image:",
-        placeholder="e.g. 'What objects do you see?' or 'Explain this diagram'",
+        "Ask about the image:",
+        placeholder="e.g. 'What is in this image?' or 'Explain this diagram'",
         key="user_prompt"
     )
     
-    if st.button("Analyze", type="primary") and image:
+    if st.button("Analyze Image", type="primary"):
         try:
             # Convert image to base64
             buffered = io.BytesIO()
@@ -115,18 +197,18 @@ with col2:
             messages = [
                 {
                     "role": "system",
-                    "content": """You are a real-time vision assistant. Analyze the current camera feed or uploaded image and:
-1. Identify objects, people, text clearly
-2. Answer follow-up questions precisely
-3. Format responses with bullet points
-4. Highlight urgent/important findings"""
+                    "content": """You are an expert vision assistant. Analyze images with:
+- Clear, structured responses
+- Bullet points for multiple objects
+- Concise explanations
+- Highlight important findings in bold"""
                 },
                 {
                     "role": "user",
                     "content": [
                         {
                             "type": "text",
-                            "text": user_prompt if user_prompt else "Describe what you see in detail"
+                            "text": user_prompt if user_prompt else "Describe this image in detail"
                         },
                         {
                             "type": "image_url",
@@ -159,13 +241,19 @@ with col2:
                     """, unsafe_allow_html=True)
         
         except Exception as e:
-            st.error(f"Error: {str(e)}")
+            st.error(f"Analysis error: {str(e)}")
 
-# Sidebar (keep your existing sidebar)
+# Sidebar
 with st.sidebar:
-    st.image("blob.png", width=200)
+    st.image("https://via.placeholder.com/200", width=200)  # Replace with your logo
     st.markdown("""
     *Powered by OpenRouter*
     """)
     st.markdown("---")
-    st.markdown("Made with ❤️ by Koshur AI")
+    st.markdown("""
+    **Tips:**
+    - For best results, use clear, well-lit images
+    - Ask specific questions for detailed answers
+    """)
+    st.markdown("Made with ❤️ by Koshur AI")
+