Update app.py

app.py

@@ -11,6 +11,7 @@ qa_pipeline = pipeline("table-question-answering", model="google/tapas-large-fin
 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."}
@@ -18,15 +19,30 @@ def ask_table(data, question):
         headers = data[0]  # First row is headers
         rows = data[1:]    # Remaining rows are data
         
+        # 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 to TAPAS-compatible format
         processed_table = [dict(zip(headers, row)) for row in rows]
         
         if not processed_table:
             return {"answer": "Table data is empty or improperly formatted."}
         
+        # Call the model to get the answer
         answers = qa_pipeline(table=processed_table, query=question)
-        return {"answer": answers.get("answer", "No answer found.")}
+        answer_text = answers.get("answer", "No answer found.")
+        
+        return {"answer": answer_text}
+    
     except Exception as e:
+        logging.error(f"Error processing table: {str(e)}")
         return {"answer": f"Error: {str(e)}"}
 
 # Define Gradio interface
@@ -38,7 +54,7 @@ iface = gr.Interface(
             row_count=(2, "dynamic"),
             col_count=(19, "dynamic"),
             type="array",
-            label="Input Table"
+            label="Input Table (First row = Headers)"
         ),
         gr.Textbox(
             lines=2,