Update app.py

app.py

@@ -1,43 +1,42 @@
+import os
 import streamlit as st
-import pandas as pd
-from groq import Groq  # Ensure Groq client library is installed
+from groq import Groq
 
-# Initialize Groq API
+# API Key Initialization
 GROQ_API_KEY = "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"
 client = Groq(api_key=GROQ_API_KEY)
 
-st.title("AI Prediction with Groq API")
-uploaded_file = st.file_uploader("Upload your mission dataset (CSV)", type=["csv"])
-
-if uploaded_file:
-    # Read the dataset
-    df = pd.read_csv(uploaded_file)
-    st.dataframe(df)
-
-    # Select mission ID for prediction
-    mission_ids = df["Mission ID"].unique()
-    selected_mission_id = st.selectbox("Select Mission ID for prediction:", mission_ids)
-
-    # Display selected mission details
-    mission_data = df[df["Mission ID"] == selected_mission_id].iloc[0]
-    st.write("Selected Mission Data:", mission_data)
-
-    if st.button("Predict Resource Optimization"):
-        # Prepare data payload
-        payload = {
-    "id": mission_data["Mission ID"],  # Keep as a string
-    "distance_from_earth": float(mission_data["Distance from Earth (light-years)"]),
-    "duration": float(mission_data["Mission Duration (years)"]),
-    "crew_size": int(mission_data["Crew Size"]),
-    "fuel_consumption": float(mission_data["Fuel Consumption (tons)"]),
-    "payload_weight": float(mission_data["Payload Weight (tons)"])
-}
-
-       
-
-        # Call the Groq API
-        try:
-            response = client.predict(payload)
-            st.success(f"Prediction Result: {response}")
-        except Exception as e:
-            st.error(f"Error: {str(e)}")
+# Function to fetch mission details by ID
+def get_mission_details(mission_id):
+    """Fetch resource optimization or mission details using Groq API."""
+    try:
+        response = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": f"Provide detailed insights about mission ID: {mission_id} in space exploration."
+                }
+            ],
+            model="llama-3.3-70b-versatile",
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"Error fetching mission details: {e}"
+
+# Streamlit App UI
+st.title("🚀 AI-Powered Space Mission Insights")
+st.sidebar.header("Mission Selection")
+
+# Mission ID input
+mission_id = st.sidebar.text_input("Enter Mission ID", "MSN-0003")
+
+if mission_id:
+    st.subheader(f"Mission Insights for ID: {mission_id}")
+    st.text("Fetching details...")
+
+    # Fetch and display mission details
+    mission_details = get_mission_details(mission_id)
+    st.write(mission_details)
+
+# Additional UI Elements
+st.sidebar.info("Enter a valid mission ID to explore detailed information!")