Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from transformers import pipeline
+import PyPDF2
+import markdown
 
 # Preload models
 models = {
@@ -20,7 +22,20 @@ def load_model(model_name):
 def answer_question(model_name, file, question):
     model = load_model(model_name)
     
-    context = file.read().decode('utf-8') if file else ""
+    if file is not None:
+        if file.type == "application/pdf":
+            pdf_reader = PyPDF2.PdfFileReader(file)
+            context = ""
+            for page_num in range(pdf_reader.numPages):
+                context += pdf_reader.getPage(page_num).extract_text()
+        elif file.name.endswith(".md"):
+            context = file.read().decode('utf-8')
+            context = markdown.markdown(context)
+        else:
+            context = file.read().decode('utf-8')
+    else:
+        context = ""
+    
     result = model(question=question, context=context)
     answer = result['answer']
     score = result['score']
@@ -35,7 +50,9 @@ with gr.Blocks() as interface:
     gr.Markdown(
         """
         # Question Answering System
-        Upload a document and ask questions to get answers based on the context.
+        Upload a document (text, PDF, or Markdown) and ask questions to get answers based on the context.
+        
+        **Supported File Types**: `.txt`, `.pdf`, `.md`
         """)
     
     with gr.Row():
@@ -46,7 +63,7 @@ with gr.Blocks() as interface:
         )
     
     with gr.Row():
-        file_input = gr.File(label="Upload Document")
+        file_input = gr.File(label="Upload Document", file_types=["text", "pdf", "markdown"])
         question_input = gr.Textbox(lines=2, placeholder="Enter your question here...", label="Question")
     
     with gr.Row():