Update app.py

app.py

@@ -1,44 +1,58 @@
-import os, re, openai, pandas as pd, traceback
+import os
+import re
+import openai
+import pandas as pd
+import traceback
 from collections import Counter, defaultdict
 from apscheduler.schedulers.background import BackgroundScheduler
 import gradio as gr
+
 from utils import (
     SECTOR, DYN, save_dynamic,
     fetch_news, keyword_hit,
     validate_ticker, send_tg, ist_now
 )
 
+# ── initialize LLM client ─────────────────────────────────────────────────
 openai.api_base = "https://api.groq.com/openai/v1"
 openai.api_key  = os.getenv("GROQ_API_KEY")
 
+# ── helper: ask LLM for rationale ──────────────────────────────────────────
 def llm_reason(theme, tickers, headlines):
-    prompt = f"""Theme: {theme}
+    prompt = f"""
+Theme: {theme}
 Sample news: {' | '.join(headlines[:4])}
 
-Write TWO crisp bullets (<60 words) explaining why {', '.join(tickers)} benefit now."""
+Write TWO crisp bullets (<60 words total) explaining why {', '.join(tickers)} benefit now.
+"""
     rsp = openai.ChatCompletion.create(
         model="llama3-70b-8192",
         messages=[{"role":"user","content":prompt}],
-        temperature=0.35, max_tokens=130
+        temperature=0.35,
+        max_tokens=130
     )
     return rsp.choices[0].message.content.strip()
 
+# ── helper: ask LLM for new tickers ────────────────────────────────────────
 def suggest_tickers(theme):
     ask = openai.ChatCompletion.create(
         model="llama3-8b-8192",
         messages=[{"role":"user","content":
-          f"List up to 5 NSE tickers that will benefit from '{theme}'. Return comma‑separated tickers only."}],
-        temperature=0.4, max_tokens=20
+            f"List up to 5 NSE tickers that will benefit from '{theme}'. Return comma-separated tickers only."
+        }],
+        temperature=0.4,
+        max_tokens=20
     )
     raw = re.split(r"[,\s]+", ask.choices[0].message.content.strip())
     return [validate_ticker(t.upper()) for t in raw]
 
+# ── core logic (may raise) ────────────────────────────────────────────────
 def _daily_job():
-    """Core logic, may raise exceptions."""
     news  = fetch_news()
     freq  = Counter()
     hits  = defaultdict(list)
 
+    # count keyword hits per theme
     for art in news:
         text = art["title"] + " " + art.get("description", "")
         for theme, cfg in SECTOR.items():
@@ -46,10 +60,10 @@ def _daily_job():
                 freq[theme] += 1
                 hits[theme].append(art["title"])
 
+    rows, cards = [], []
     if not freq:
-        return pd.DataFrame([], columns=["theme","hits","stocks","reason"])
+        return pd.DataFrame(rows, columns=["theme","hits","stocks","reason"])
 
-    rows, cards = [], []
     for theme, count in freq.most_common():
         base  = SECTOR[theme]["tickers"]
         extra = DYN.get(theme, [])
@@ -64,41 +78,43 @@ def _daily_job():
 
         cards.append(
             f"*{theme}* ({count} hits)\n"
-            f"Stocks: {', '.join('`'+t[:-3]+'`' for t in tickers)}\n{reason}"
+            f"Stocks: {', '.join('`'+t[:-3]+'`' for t in tickers)}\n"
+            f"{reason}"
         )
-        rows.append({"theme":theme, "hits":count,
-                     "stocks":', '.join(tickers), "reason":reason})
+        rows.append({
+            "theme": theme,
+            "hits": count,
+            "stocks": ', '.join(tickers),
+            "reason": reason
+        })
 
     header = f"📈 *Macro Theme Brief*  {ist_now().strftime('%d %b %Y %H:%M')}"
     send_tg(header + "\n\n" + "\n\n".join(cards))
     return pd.DataFrame(rows)
 
+# ── safe wrapper (catches & displays errors) ──────────────────────────────
 def daily_job_safe():
     try:
-        result = _daily_job()
-        return result
-    except Exception as e:
+        return _daily_job()
+    except Exception:
         tb = traceback.format_exc()
-        # Log to Space logs
         print("Error in daily_job:", tb)
-        # Show traceback in UI
-        return f"```python\n{tb}\n```"
+        return "```python\n" + tb + "\n```"
 
-# ── Scheduler ──────────────────────────────────────────────────────────────
+# ── schedule to run at 08:00 IST (02:30 UTC) ───────────────────────────────
 sched = BackgroundScheduler(timezone="UTC")
-sched.add_job(daily_job_safe, trigger="cron", hour=2, minute=30)  # 08:00 IST
+sched.add_job(daily_job_safe, trigger="cron", hour=2, minute=30)
 sched.start()
 
-# ── Gradio UI ──────────────────────────────────────────────────────────────
+# ── Gradio UI ─────────────────────────────────────────────────────────────
 if __name__ == "__main__":
-    import os
     port = int(os.environ.get("PORT", 7860))
     with gr.Blocks() as demo:
-        gr.Markdown("### 🇮🇳 Macro‑Theme Bot  (Elec MFG | Semicon | H₂ | Ethanol | RE)")
+        gr.Markdown("### 🇮🇳 Macro‑Theme Bot (Elec MFG | Semicon | H₂ | Ethanol | RE)")
         out = gr.Textbox(lines=20, label="Output / Error")
         gr.Button("Run now").click(fn=daily_job_safe, outputs=out)
     demo.launch(
         server_name="0.0.0.0",
         server_port=port,
-        ssr_mode=False,
+        ssr_mode=False
     )