Update app.py

app.py

@@ -1,12 +1,14 @@
+# app.py
+
 import gradio as gr
 import pandas as pd
 from transformers import pipeline
 
-# Load & stringify your CSV
+# 1) Load & stringify your CSV
 df = pd.read_csv("synthetic_profit.csv")
 table = df.astype(str).to_dict(orient="records")
 
-# Instantiate TAPAS pipeline
+# 2) TAPAS pipeline
 qa = pipeline(
     "table-question-answering",
     model="google/tapas-base-finetuned-wtq",
@@ -14,42 +16,48 @@ qa = pipeline(
     device=-1
 )
 
-# Four + one few-shot examples
+# 3) Instruction + few-shot examples
+PREFIX = """
+You are a table-QA assistant.  
+- When the question asks for “total” or “sum” of a column:  
+    • Filter rows as specified.  
+    • Compute the sum of that column.  
+    • Return exactly one number (the sum).  
+- When the question asks for “average” or “mean”:  
+    • Filter rows as specified.  
+    • Compute the mean.  
+    • Return exactly one number (the mean).  
+"""
+
 EXAMPLES = """
 Example 1:
 Q: What is the total revenue for Product A in EMEA in Q1 2024?
 A: Filter Product=A & Region=EMEA & FiscalYear=2024 & FiscalQuarter=Q1, then sum Revenue → 3075162.49
 
 Example 2:
-Q: What is the total cost for Product A in EMEA in Q1 2024?
-A: Filter Product=A & Region=EMEA & FiscalYear=2024 & FiscalQuarter=Q1, then sum Cost → 2894321.75
-
-Example 3:
-Q: What is the total margin for Product A in EMEA in Q1 2024?
-A: Filter Product=A & Region=EMEA & FiscalYear=2024 & FiscalQuarter=Q1, then sum ProfitMargin → 0.18
-
-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
+A: Filter Product=A & FiscalYear=2024 & FiscalQuarter=Q1, then sum Revenue → 12032469.96
 """
 
 def answer_question(question: str) -> str:
-    prompt = EXAMPLES + f"\nQ: {question}\nA:"
-    out = qa(table=table, query=prompt)
-    return out.get("answer", "No answer found.")
-
+    prompt = PREFIX + EXAMPLES + f"\nQ: {question}\nA:"
+    try:
+        out = qa(table=table, query=prompt)
+        return out.get("answer", "No answer found.")
+    except Exception as e:
+        return f"❌ Pipeline error:\n{e}"
+
+# 4) Gradio UI
 iface = gr.Interface(
     fn=answer_question,
-    inputs=gr.Textbox(lines=2, placeholder="Ask a question…"),
-    outputs=gr.Textbox(lines=3),
+    inputs=gr.Textbox(lines=2, placeholder="e.g. What is the total revenue for Product A in EMEA in Q1 2024?"),
+    outputs=gr.Textbox(lines=2),
     title="SAP Profitability Q&A",
-    description="TAPAS few-shot sum/mean demo",
+    description=(
+        "Table‐QA with TAPAS: instructions + examples force a single-number sum/mean output."
+    ),
     allow_flagging="never",
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(server_name="0.0.0.0", server_port=7860)