Upload app.py

app.py

@@ -300,9 +300,11 @@ Two tabs: Price Prediction (Finnhub with Alpha Vantage fallback) and Equity Rese
 					try:
 						candles = fetch_alpha_vantage_series_daily(sym, outputsize="compact")
 					except Exception as e2:
-						return f"Error fetching candles: {e2}"
+						yield f"Error fetching candles: {e2}"
+						return
 				prompt = build_price_prediction_prompt(sym, candles)
-				return generate_response_stream(prompt, temperature=temperature, max_new_tokens=int(max_tokens))
+				for text in generate_response_stream(prompt, temperature=temperature, max_new_tokens=int(max_tokens)):
+					yield text
 
 			btn.click(on_predict, inputs=[symbol, resolution, count, temp, max_new], outputs=out, show_progress=True)
 
@@ -316,10 +318,12 @@ Two tabs: Price Prediction (Finnhub with Alpha Vantage fallback) and Equity Rese
 			def on_report(sym, temperature, max_tokens):
 				try:
 					overview = fetch_alpha_vantage_overview(sym)
-					prompt = build_equity_research_prompt(sym, overview)
-					return generate_response_stream(prompt, temperature=temperature, max_new_tokens=int(max_tokens))
 				except Exception as e:
-					return f"Error generating report: {e}"
+					yield f"Error fetching fundamentals: {e}"
+					return
+				prompt = build_equity_research_prompt(sym, overview)
+				for text in generate_response_stream(prompt, temperature=temperature, max_new_tokens=int(max_tokens)):
+					yield text
 
 			btn2.click(on_report, inputs=[symbol2, temp2, max_new2], outputs=out2, show_progress=True)