Update app.py

app.py

@@ -88,6 +88,24 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=800):
     # ✅ Keep only last 10 exchanges to prevent unnecessary repetition
     return response, chat_history[-10:]
 
+# ✅ NASA API Function to get space data
+def get_nasa_data(query):
+    try:
+        if "apod" in query.lower() or "picture of the day" in query.lower():
+            response = requests.get(f"https://api.nasa.gov/planetary/apod?api_key={NASA_API_KEY}")
+            if response.status_code == 200:
+                data = response.json()
+                return {
+                    "title": data.get("title", "NASA Image"),
+                    "date": data.get("date", ""),
+                    "explanation": data.get("explanation", ""),
+                    "url": data.get("url", "")
+                }
+        return None
+    except Exception as e:
+        print(f"Error fetching NASA data: {e}")
+        return None
+
 # ✅ Streamlit UI
 st.title("🚀 HAL - NASA AI Assistant")
 
@@ -109,7 +127,7 @@ st.markdown("""
         background-color: #2196F3;
         border: none;
         color: white;
-        padding: 5px 11px;
+        padding: 5px 10px;
         text-align: center;
         text-decoration: none;
         display: inline-block;
@@ -347,4 +365,59 @@ st.markdown("""
 
 # Add voice status indicator
 st.markdown("""
-    <div class="status
+    <div class="status-bar">
+        <span id="voice-status">Initializing voice recognition...</span>
+        <span id="voice-indicator" class="voice-indicator" onclick="toggleVoiceRecognition()"></span>
+    </div>
+""", unsafe_allow_html=True)
+
+# Regular text input
+user_input = st.chat_input("Type your message here or just speak...")
+
+# Hidden input for speech results
+speech_result = st.text_input("Speech Result", key="speech_input", label_visibility="collapsed")
+
+# Hidden button to submit speech
+st.markdown('<button id="submit-speech" style="display:none;">Submit Speech</button>', unsafe_allow_html=True)
+
+# Auto-speak toggle
+st.checkbox("Auto-speak responses", value=st.session_state.auto_speak, key="auto_speak_toggle", 
+            on_change=lambda: setattr(st.session_state, "auto_speak", st.session_state.auto_speak_toggle))
+
+# Display chat history
+for i, msg in enumerate(st.session_state.chat_history):
+    if msg["role"] == "user":
+        st.markdown(f'<div class="container"><div class="user-msg">You: {msg["content"]}</div></div>', unsafe_allow_html=True)
+    else:
+        msg_id = f"msg-{i}"
+        st.markdown(f'<div class="container"><div class="assistant-msg">HAL: <span id="{msg_id}">{msg["content"]}</span> <button class="speak-button" onclick="speakText(document.getElementById(\'{msg_id}\').textContent)">🔊</button></div></div>', unsafe_allow_html=True)
+
+# Process user input
+if user_input or speech_result:
+    # Prioritize speech result if available
+    query = speech_result if speech_result else user_input
+    
+    # Get NASA data if applicable
+    nasa_data = get_nasa_data(query)
+    
+    # Generate response
+    system_message = "You are HAL, an AI assistant specialized in NASA and space knowledge. Provide concise, factual responses."
+    response, st.session_state.chat_history = get_response(system_message, st.session_state.chat_history, query)
+    
+    # Display NASA image if available
+    if nasa_data:
+        st.image(nasa_data["url"], caption=f"{nasa_data['title']} - {nasa_data['date']}")
+        st.write(nasa_data["explanation"])
+    
+    # Force a rerun to update the chat display
+    st.rerun()
+
+# Add JavaScript to ensure the page scrolls to the bottom on new messages
+st.markdown("""
+<script>
+    // Scroll to bottom of page on load
+    window.onload = function() {
+        window.scrollTo(0, document.body.scrollHeight);
+    }
+</script>
+""", unsafe_allow_html=True)