added improvements

app.py

@@ -22,15 +22,18 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
+def validate_ticker(ticker):
+    # Ensure ticker is uppercase and length is reasonable (1-5 characters)
+    return ticker.strip().upper()
+
 
 def fetch_stock_data(ticker):
-    print("fetching stock data")
+    ticker = validate_ticker(ticker)  # Validate and clean input
     stock = yf.Ticker(ticker)
     hist = stock.history(period="1mo")
     return hist.tail(5)
 
 def fetch_stock_news(ticker, NEWSAPI_KEY):
-    print("fetching stock news")
     api_url = f"https://newsapi.org/v2/everything?q={ticker}&apiKey={NEWSAPI_KEY}"
     response = requests.get(api_url)
     if response.status_code == 200:
@@ -40,7 +43,6 @@ def fetch_stock_news(ticker, NEWSAPI_KEY):
         return [{"error": "Unable to fetch news."}]
 
 def calculate_moving_average(ticker, window=5):
-    print("calculating moving average")
     stock = yf.Ticker(ticker)
     hist = stock.history(period="1mo")
     hist[f"{window}-day MA"] = hist["Close"].rolling(window=window).mean()
@@ -51,7 +53,7 @@ llm = HuggingFacePipeline(pipeline=pipe)
 stock_data_tool = Tool(
     name="Stock Data Fetcher",
     func=fetch_stock_data,
-    description="Fetch recent stock data for a given ticker."
+    description="Fetch recent stock data for a valid stock ticker (e.g., AAPL for Apple)."
 )
 
 stock_news_tool = Tool(
@@ -73,7 +75,7 @@ agent = initialize_agent(
     llm=llm,
     agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
     verbose=True,
-    handle_parsing_errors=True
+    handle_parsing_errors=True  # Enables automatic handling of parsing issues
 )
 
 print(fetch_stock_data("AAPL"))
@@ -87,9 +89,11 @@ query = st.text_input("Enter your query:")
 
 if st.button("Submit"):
     if query:
+        st.write("Debug: User Query ->", query)
         with st.spinner("Processing..."):
             try:
                 response = agent.run(query)
+                st.write("Debug: Agent Response ->", response)
                 st.success("Response:")
                 st.write(response)
             except Exception as e: