Update app.py

app.py

@@ -39,17 +39,22 @@ def model(model_name):
 def qa_result(context, question, file_path):
     model_name = "timpal0l/mdeberta-v3-base-squad2"
     pipe = model(model_name)
-
-    if len(context) == 0 and len(question) == 0:
-        text = "Որպեսզի ես կարողանամ քեզ օգնել, ինձ պիտի տրամադրես համապատասխան տեքստն ու հարցերը։"
+    if file_path is not None:
+        context = extract_text_from_pdf(file_path)
+        result = pipe(question=question, context=context)
+        answered = result['answer']
+        text = remove_references(answered)
+        
+    elif len(context) == 0 and len(question) == 0:: 
+            text = "Որպեսզի ես կարողանամ քեզ օգնել, ինձ պիտի տրամադրես համապատասխան տեքստն ու հարցերը։"
     elif len(context) == 0:
         text = "Ես չեմ կարողանամ քեզ օգնել եթե ինձ չտրամադրես տեքստը"
     elif len(question) == 0:
         text = "Ես չեմ կարողանամ քեզ օգնել եթե ինձ չտաս հարցդ"
     else:
-        if file_path is not None:
-            # File was uploaded, extract text from the file
-            context = extract_text_from_pdf(file_path)
+        # if file_path is not None:
+        #     # File was uploaded, extract text from the file
+        #     context = extract_text_from_pdf(file_path)
         # else: No file uploaded, use the provided context as is
         result = pipe(question=question, context=context)
         answered = result['answer']