Update app.py

app.py

@@ -346,22 +346,23 @@ def generate_final_response(prompt, history):
             print("Final Ticker: ", ticker)
             print("Getting into raw_news()")
             try:
-                context_file, ticker_stats, ticker_financials, reports, news_link = raw_news(prompt, chat_completion_params["todays_news_flag"], ticker)
+                if not chat_completion_params['generic_query']:
+                    context_file, ticker_stats, ticker_financials, reports, news_link = raw_news(prompt, chat_completion_params["todays_news_flag"], ticker)
+                
                 print(f"Generating response for **{prompt}**")
                 env = Environment(loader=FileSystemLoader("templates/"), autoescape=True)
                 # env.globals['include'] = lambda filename: env.loader.get_source(env, filename)[0]
                 template = env.get_template("system_prompt.txt")
-                for response_type in ["Response-1"]:
-                    content = template.render(todays_date=todays_date,ticker_financials=ticker_financials,response_type=response_type,chat_completion_params=chat_completion_params,context_file=context_file,question=question,ticker=ticker, ticker_stats = ticker_stats, reports=reports, news_link=news_link)
-                    print(content)
-                    output=""
-                    try:
-                        stream = client.text_generation(content, **generate_kwargs, stream=True, details=True, return_full_text=True)
-                        for response in stream:
-                            output += response.token.text
-                            yield output
-                    except StopAsyncIteration:
-                        yield "Sorry, could you provide more details to clarify your query"
+                content = template.render(todays_date=todays_date,ticker_financials=ticker_financials or None,response_type="Response-1",chat_completion_params=chat_completion_params,context_file=context_file or None,question=question,ticker=ticker, ticker_stats = ticker_stats or None, reports=reports or None, news_link=news_link or None)
+                print(content)
+                output=""
+                try:
+                    stream = client.text_generation(content, **generate_kwargs, stream=True, details=True, return_full_text=True)
+                    for response in stream:
+                        output += response.token.text
+                        yield output
+                except StopAsyncIteration:
+                    yield "Sorry, could you provide more details to clarify your query"
             except:
                 yield "Sorry, your query couldn't be processed. Retry with correct name of stock"
     except: