Update app.py

app.py

@@ -2,19 +2,19 @@ import gradio as gr
 import pandas as pd
 from transformers import pipeline
 
-# 1) Load & stringify your CSV
+# Load & stringify your CSV
 df = pd.read_csv("synthetic_profit.csv")
 table = df.astype(str).to_dict(orient="records")
 
-# 2) Instantiate the TAPAS pipeline from Transformers
+# Instantiate TAPAS pipeline
 qa = pipeline(
     "table-question-answering",
     model="google/tapas-base-finetuned-wtq",
     tokenizer="google/tapas-base-finetuned-wtq",
-    device=-1,   # CPU; change to 0 if you have a GPU
+    device=-1
 )
 
-# 3) Few-shot examples teach “filter + sum” vs. “filter + mean”
+# Four + one few-shot examples
 EXAMPLES = """
 Example 1:
 Q: What is the total revenue for Product A in EMEA in Q1 2024?
@@ -31,28 +31,25 @@ A: Filter Product=A & Region=EMEA & FiscalYear=2024 & FiscalQuarter=Q1, then sum
 Example 4:
 Q: What is the average profit margin for Product A in EMEA in Q1 2024?
 A: Filter Product=A & Region=EMEA & FiscalYear=2024 & FiscalQuarter=Q1, then mean ProfitMargin → 0.18
+
+Example 5:
+Q: What is the total revenue for Product A in Q1 2024?
+A: Filter Product=A & FiscalYear=2024 & FiscalQuarter=Q1, then sum Revenue → YOUR_SUM_HERE
 """
 
 def answer_question(question: str) -> str:
     prompt = EXAMPLES + f"\nQ: {question}\nA:"
-    try:
-        result = qa(table=table, query=prompt)
-        return result.get("answer", "No answer found.")
-    except Exception as e:
-        return f"❌ Pipeline error:\n{e}"
+    out = qa(table=table, query=prompt)
+    return out.get("answer", "No answer found.")
 
-# 4) Gradio UI
 iface = gr.Interface(
     fn=answer_question,
-    inputs=gr.Textbox(lines=2, placeholder="e.g. What is the total revenue for Product A in Q1 2024?"),
+    inputs=gr.Textbox(lines=2, placeholder="Ask a question…"),
     outputs=gr.Textbox(lines=3),
     title="SAP Profitability Q&A",
-    description=(
-        "Ask simple sum/mean questions on the synthetic SAP data.  \n"
-        "Powered by google/tapas-base-finetuned-wtq with four few-shot examples."
-    ),
+    description="TAPAS few-shot sum/mean demo",
     allow_flagging="never",
 )
 
 if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    iface.launch()