Update app.py

app.py

@@ -3,11 +3,11 @@ import requests
 import re
 import time
 
-# --- 🔑 API KEYS (replace with env vars or Hugging Face secrets in production) ---
+# --- 🔑 API KEYS ---
 TOGETHER_API_KEY = "tgp_v1_ZytvDbMu9PMwIlnBZEfYSq9nzJAYwS0MecjY9Kt7RxE"
 SERPER_API_KEY = "75f06519187851ad63486c3012b34c5e0e6501f1"
 
-# --- 💬 LLM: Generate Structured Startup Ideas ---
+# --- 💬 Generate Startup Ideas ---
 def generate_startup_ideas(prompt, retries=3, delay=2):
     url = "https://api.together.xyz/v1/completions"
     headers = {
@@ -15,7 +15,7 @@ def generate_startup_ideas(prompt, retries=3, delay=2):
         "Content-Type": "application/json"
     }
     data = {
-        "model": "mistralai/Mistral-7B-Instruct",
+        "model": "mistralai/Mistral-7B-Instruct-v0.1",
         "prompt": f"""
 Suggest 3 unique startup ideas based on this interest: \"{prompt}\".
 Each idea should be short, clear, and numbered like this:
@@ -23,22 +23,28 @@ Each idea should be short, clear, and numbered like this:
 2. ...
 """,
         "max_tokens": 300,
-        "temperature": 0.7
+        "temperature": 0.7,
+        "top_p": 0.95
     }
 
     for attempt in range(retries):
         try:
             response = requests.post(url, headers=headers, json=data)
             result = response.json()
-            print("API response:", result)
 
             if "choices" in result and result["choices"]:
                 raw_text = result["choices"][0].get("text", "").strip()
                 ideas = re.findall(r"\d+\.\s*(.*?):\s*(.*)", raw_text)
-                return [f"{i+1}. {title.strip()}: {desc.strip()}" for i, (title, desc) in enumerate(ideas)]
+                if ideas:
+                    return [f"{i+1}. {title.strip()}: {desc.strip()}" for i, (title, desc) in enumerate(ideas)]
+                else:
+                    print("⚠️ No valid ideas format. Raw output:", raw_text)
+
+            else:
+                print("⚠️ Unexpected API response:", result)
 
         except Exception as e:
-            print(f"[Retry {attempt+1}] Error:", e)
+            print(f"[Attempt {attempt+1}] Error: {e}")
 
         time.sleep(delay)
 
@@ -55,23 +61,24 @@ def market_research(query):
         return [r["title"] + " - " + r["link"] for r in results.get("organic", [])[:5]]
     except Exception as e:
         print("Market research error:", e)
-        return ["Failed to fetch market research."]
+        return ["❌ Market research failed."]
 
-# --- 🌐 STREAMLIT UI ---
+# --- 🌐 Streamlit UI ---
 st.set_page_config(page_title="Startup Co-Founder Agent", layout="centered")
 st.title("🚀 Startup Co-Founder Agent")
-st.write("Get startup ideas + instant market research with free AI tools!")
+st.write("Get startup ideas + instant market research. Free & AI-powered.")
 
 user_prompt = st.text_input("What are you interested in building a startup around?", "AI for education")
 
 if st.button("Generate Startup Ideas"):
     if not TOGETHER_API_KEY or not SERPER_API_KEY:
-        st.error("API keys are not set. Please configure them.")
+        st.error("API keys are missing. Set them in environment or Hugging Face Secrets.")
     else:
-        with st.spinner("Cooking up ideas and doing some Googling... 🍳"):
+        with st.spinner("Brewing ideas with GPUs and coffee... ☕🤖"):
             ideas = generate_startup_ideas(user_prompt)
+
             if not ideas:
-                st.error("Still no ideas. Please try again later!")
+                st.error("Still no ideas. Maybe try again in a few seconds.")
             else:
                 st.subheader("💡 Startup Ideas")
                 for idea in ideas: