Update app.py

app.py

@@ -29,7 +29,7 @@ sql_gen = pipeline(
 )
 
 def answer_profitability(question: str) -> str:
-    # 1) Ask TAPEX to write SQL
+    # 1) Generate SQL
     prompt = (
         f"-- Translate to SQL for table `sap` ({schema})\n"
         f"Question: {question}\n"
@@ -37,8 +37,26 @@ def answer_profitability(question: str) -> str:
     )
     sql = sql_gen(prompt)[0]['generated_text'].strip()
 
-    # 2) Try to run it
+    # 2) Try to execute it
     try:
         df_out = con.execute(sql).df()
     except Exception as e:
-        return f"""**❌ SQL Error**  
+        # Use a normal f-string with explicit \n for newlines
+        return (
+            f"❌ **SQL Error**\n"
+            f"```\n{e}\n```\n\n"
+            f"**Generated SQL**\n"
+            f"```sql\n{sql}\n```"
+        )
+
+    # 3) Format successful result
+    if df_out.empty:
+        return (
+            "No rows returned.\n\n"
+            f"**Generated SQL**\n```sql\n{sql}\n```"
+        )
+
+    if df_out.shape == (1,1):
+        return str(df_out.iat[0,0])
+
+    return df_out.to_markdown(index=False)