Update app.py

app.py

@@ -126,6 +126,26 @@ def ask(question, ids, language, request: gr.Request):
                           {'role': 'assistant', 'content': result_markdown(result)}])
         return item.chat, ''
 
+@spaces.GPU
+def ask(question, ids, language, request: gr.Request):
+    item = session(request)
+    with item.lock:
+        if not question.strip():
+            return item.chat, ''
+        result = item.engine.present(item.engine.ask(question, ids or []), language)
+        item.chat.extend([{'role': 'user', 'content': question},
+                          {'role': 'assistant', 'content': result_markdown(result)}])
+        return item.chat, ''
+
+
+@spaces.GPU
+def summarize(ids, language, request: gr.Request, progress=gr.Progress()):
+    item = session(request)
+    with item.lock:
+        if not ids:
+            return 'Upload and select a PDF first.'
+        sections = []
+
 
 def summarize(ids, language, request: gr.Request, progress=gr.Progress()):
     item = session(request)
@@ -206,6 +226,5 @@ def build_app():
 
 if __name__ == '__main__':
     logging.info('LexQuery Gemini model: %s', GEMINI_MODEL)
-    # Changed default host to 0.0.0.0 so Hugging Face Spaces can expose it
     build_app().launch(server_name=os.getenv('GRADIO_SERVER_NAME', '0.0.0.0'),
                        server_port=int(os.getenv('GRADIO_SERVER_PORT', '7860')), share=False)