Update app.py

app.py

@@ -58,72 +58,46 @@ def model(model_name):
 model_name = "timpal0l/mdeberta-v3-base-squad2"
 pipe = model(model_name)
 
-# def qa_result(context, question, file, pipe = pipe):  
-#     if file is not None:
-#         allowed_types = [".pdf", ".txt", ".docx"]
-#         extension = "." + file.name.split(".")[-1].lower()
-#         if not extension in allowed_types:
-#             text = "Խնդրում եմ ներբեռնել .pdf, .txt, կամ .docx տիպի ֆայլեր։"
-#         else:
-#             if extension == allowed_types[0]:
-#                 context = extract_text_from_pdf(file.name)
-#             elif extension == allowed_types[1]:
-#                 context = extract_text_from_txt(file.name)
-#             else:
-#                 context = extract_text_from_doc(file.name)
+def qa_result(context, question, file, pipe = pipe):  
+    if file is not None:
+        allowed_types = [".pdf", ".txt", ".docx"]
+        extension = "." + file.name.split(".")[-1].lower()
+        if not extension in allowed_types:
+            text = "Խնդրում եմ ներբեռնել .pdf, .txt, կամ .docx տիպի ֆայլեր։"
+        else:
+            if extension == allowed_types[0]:
+                context = extract_text_from_pdf(file.name)
+            elif extension == allowed_types[1]:
+                context = extract_text_from_txt(file.name)
+            else:
+                context = extract_text_from_doc(file.name)
                 
-#             result = pipe(question=question, context=context)
-#             answered = result['answer']
-#             text = remove_references(answered)
+            # result = pipe(question=question, context=context)
+            # answered = result['answer']
+            # text = remove_references(answered)
    
-#     elif file is None and len(context) == 0 and len(question) == 0:
-#         text = "Որպեսզի ես կարողանամ քեզ օգնել, ինձ պիտի տրամադրես համապատասխան տեքստն կամ ֆայլն ու հարցերը։"
+    elif file is None and len(context) == 0 and len(question) == 0:
+        text = "Որպեսզի ես կարողանամ քեզ օգնել, ինձ պիտի տրամադրես համապատասխան տեքստն կամ ֆայլն ու հարցերը։"
    
-#     else:
-#         if len(context) == 0 and len(question) == 0:
-#             text = "Որպեսզի ես կարողանամ քեզ օգնել, ինձ պիտի տրամադրես համապատասխան տեքստն ու հարցերը։"
-#         elif len(context) == 0:
-#             text = "Ես չեմ կարողանա քեզ օգնել եթե ինձ չտրամադրես տեքստը"
-#         elif len(context) == 0:
-#             text = "Ես չեմ կարողանա քեզ օգնել եթե ինձ չտաս հարցդ"
-#         else:
-#             result = pipe(question=question, context=context)
-#             answered = result['answer']
-#             text = remove_references(answered)
-#     text = text.replace('(', '', 1)
-#     text = text.replace(',', '', len(text)-1)
-            
-#     return text.capitalize()
-
-
-def qa_result(context, question, file_name, file_extension, pipe=pipe):
-    text = ""
-
-    if len(context) == 0 or len(question) == 0:
-        text = "Որպեսզի ես կարողանամ քեզ օգնել, ինձ պիտի տրամադրես համապատասխան տեքստը և հարցերը։"
     else:
-        if file_extension not in [".pdf", ".txt", ".docx"]:
-            text = "Խնդրում եմ ներբեռնել .pdf, .txt, կամ .docx տիպի ֆայլեր։"
-        else:
-            if file_extension == ".pdf":
-                context = extract_text_from_pdf(file_name)
-            elif file_extension == ".txt":
-                context = extract_text_from_txt(file_name)
-            else:
-                context = extract_text_from_doc(file_name)
-
-            result = pipe(question=question, context=context)
-            answered = result['answer']
-            text = remove_references(answered)
-
-    text = text[1:]  # Remove the first parenthesis
-    if text.endswith(','):
-        text = text[:-1]  # Remove trailing comma
-
-    
+        if len(context) == 0 and len(question) == 0:
+            text = "Որպեսզի ես կարողանամ քեզ օգնել, ինձ պիտի տրամադրես համապատասխան տեքստն ու հարցերը։"
+        elif len(context) == 0:
+            text = "Ես չեմ կարողանա քեզ օգնել եթե ինձ չտրամադրես տեքստը"
+        elif len(context) == 0:
+            text = "Ես չեմ կարողանա քեզ օգնել եթե ինձ չտաս հարցդ"
+        # else:
+            
+    result = pipe(question=question, context=context)
+    answered = result['answer']
+    text = remove_references(answered)
+    text = text.replace('(', '', 1)
+    text = text.replace(',', '', len(text)-1)
+            
     return text.capitalize()
 
 
+
 theme = gr.themes.Soft().set(
     body_background_fill='*background_fill_secondary',
     body_text_color_subdued='*body_text_color',