Update app.py

app.py

@@ -1,3 +1,4 @@
+does my currect app.pu able to do it?
 from dotenv import load_dotenv
 import os
 import gradio as gr
@@ -73,41 +74,7 @@ def explain_sql_output(sql_query, query_result):
     return chat_completion.choices[0].message.content.strip()
 
 
-### --- TAB 3: SQL Follow-Ups --- ###
-def followup_prompt(sql_query, query_result):
-    system_prompt = """
-You are an assistant that suggests insightful follow-up questions based on SQL query output.
-
-Instructions:
-- Use the SQL query and result to infer possible next questions.
-- Output 3 to 5 follow-up questions.
-- Be helpful, curious, and relevant to the data.
-- Do NOT repeat the explanation. Only return follow-up questions.
-- Format as a list.
-"""
-    user_prompt = f"""
-SQL Query:
-{sql_query}
-Query Result:
-{query_result}
-
-Follow-Up Questions:
-"""
-    return system_prompt.strip(), user_prompt.strip()
-
-def generate_followups(sql_query, query_result):
-    system_prompt, user_prompt = followup_prompt(sql_query, query_result)
-    chat_completion = client.chat.completions.create(
-        messages=[
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": user_prompt}
-        ],
-        model="llama3-70b-8192"
-    )
-    return chat_completion.choices[0].message.content.strip()
-
-
-### --- Gradio Interfaces --- ###
+### --- Gradio Interface --- ###
 tab1 = gr.Interface(
     fn=response,
     inputs=gr.JSON(label="Input JSON (question, schema)"),
@@ -127,18 +94,7 @@ tab2 = gr.Interface(
     description="Input a SQL query and its result. Get an AI-generated explanation."
 )
 
-tab3 = gr.Interface(
-    fn=generate_followups,
-    inputs=[
-        gr.Textbox(label="SQL Query"),
-        gr.Textbox(label="SQL Output (Raw JSON or Table Result)")
-    ],
-    outputs="text",
-    title="Follow-Up Generator (Groq + LLaMA3)",
-    description="Suggests follow-up questions based on a SQL query and its result."
-)
-
-demo = gr.TabbedInterface([tab1, tab2, tab3], ["SQL Generator", "Explain Output", "SQL Follow-Ups"])
+demo = gr.TabbedInterface([tab1, tab2], ["SQL Generator", "Explain Output"])
 
 if __name__ == '__main__':
-    demo.launch()
+    demo.launch()