Update app.py

app.py

@@ -9,41 +9,30 @@ logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(
 qa_pipeline = pipeline("table-question-answering", model="google/tapas-large-finetuned-wtq")
 
 def ask_table(data, question):
-    logging.info(f"Received Data: {data}")
-    logging.info(f"Received Question: {question}")
-    
     try:
-        if not data or not question:
-            return {"answer": "Please provide both a table and a question."}
+        logging.info(f"Received table: {data}, question: {question}")
         
-        headers = data[0]  # First row is headers
-        rows = data[1:]    # Remaining rows are data
+        if not data or not question:
+            return "Please provide both a table and a question."
         
-        # Check if headers and rows exist
-        if not headers or not rows:
-            return {"answer": "Table data is empty or improperly formatted."}
-
-        # Check table size (up to 50 rows, 20 columns)
-        if len(rows) > 50:
-            return {"answer": "The table has too many rows. Limit: 50 rows."}
-        if len(headers) > 20:
-            return {"answer": "The table has too many columns. Limit: 20 columns."}
+        # Convert the table to a format compatible with TAPAS
+        headers = data[0]  # Assume the first row contains headers
+        rows = data[1:]    # Remaining rows are table data
         
-        # Convert to TAPAS-compatible format
-        processed_table = [dict(zip(headers, row)) for row in rows]
+        # Process the table into dictionary format
+        processed_table = [dict(zip(headers, row)) for row in rows if any(row)]
         
         if not processed_table:
-            return {"answer": "Table data is empty or improperly formatted."}
+            return "The table is empty or invalid."
         
-        # Call the model to get the answer
+        # Query the TAPAS model
         answers = qa_pipeline(table=processed_table, query=question)
-        answer_text = answers.get("answer", "No answer found.")
+        logging.info(f"Answer: {answers}")
         
-        return {"answer": answer_text}
-    
+        return answers.get("answer", "No answer found.")
     except Exception as e:
-        logging.error(f"Error processing table: {str(e)}")
-        return {"answer": f"Error: {str(e)}"}
+        logging.error(f"Error: {str(e)}")
+        return f"Error processing your request: {str(e)}"
 
 # Define Gradio interface
 iface = gr.Interface(
@@ -54,7 +43,7 @@ iface = gr.Interface(
             row_count=(2, "dynamic"),
             col_count=(19, "dynamic"),
             type="array",
-            label="Input Table (First row = Headers)"
+            label="Input Table"
         ),
         gr.Textbox(
             lines=2,