Update app.py

app.py

@@ -3,7 +3,6 @@ import base64
 from gpt_reader.pdf_reader import PaperReader
 from gpt_reader.prompt import BASE_POINTS
 
-
 class GUI:
     def __init__(self):
         self.api_key = ""
@@ -19,67 +18,56 @@ class GUI:
         return self.session.question(question)
 
 
-
-with open("./logo.png", "rb") as f:
-    image_data = f.read()
-    image_base64 = base64.b64encode(image_data).decode("utf-8")
-    
-title = f"""
-    <h2 style="background-image: linear-gradient(to right, #3A5FCD, #87CEFA); -webkit-background-clip: text;
-        -webkit-text-fill-color: transparent; text-align: center;">
-        Speech Emotion Recognition
-    </h2>
+with gr.Blocks() as demo:
+     with open("./logo.png", "rb") as f:
+        image_data = f.read()
+        image_base64 = base64.b64encode(image_data).decode("utf-8")
+    title = f"""
+        <h2 style="background-image: linear-gradient(to right, #3A5FCD, #87CEFA); -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent; text-align: center;">
+            Speech Emotion Recognition
+        </h2>
     """
-
-description = f"""
-<div style="display: flex; align-items: center; justify-content: center; flex-direction: column;">
-    <p style="font-size: 18px; color: #4AAAFF; text-align: center;">
+    description = f"""
+    <div style="display: flex; align-items: center; justify-content: center; flex-direction: column;">
+        <p style="font-size: 18px; color: #4AAAFF; text-align: center;">
         Extract the emotion and tone, Simply upload the audio or text!
-    </p>
+        </p>
     <div style="display: flex; align-items: center; margin-bottom: 0px;">
-        <img src='data:image/jpeg;base64,{image_base64}' width='50' height='30' style="margin-right: 5px;"/>
-        <p style="font-size: 14px; color: #555;">
+            <img src='data:image/jpeg;base64,{image_base64}' width='50' height='30' style="margin-right: 5px;"/>
+            <p style="font-size: 14px; color: #555;">
             Disclaimer: This web app is for demonstration purposes only and not intended for commercial use. Contact: [email protected] for full solution.
-        </p>
+            </p>
+        </div>
     </div>
-</div>
-"""
-    
-
-with gr.Tab("Upload PDF File"):
-    pdf_input = gr.File(label="PDF File")
-    api_input = gr.Textbox(label="OpenAI API Key")
-    result = gr.Textbox(label="PDF Summary")
-    upload_button = gr.Button("Start Analyse")
-with gr.Tab("Ask question about your PDF"):
-    question_input = gr.Textbox(label="Your Question", placeholder="Authors of this paper?")
-    answer = gr.Textbox(label="Answer")
-    ask_button = gr.Button("Ask")
-with gr.Accordion("About this project"):
+    """
     gr.Markdown(
-    """## CHATGPT-PAPER-READER📝 
-        This repository provides a simple interface that utilizes the gpt-3.5-turbo 
-        model to read academic papers in PDF format locally. You can use it to help you summarize papers, 
-        create presentation slides, or simply fulfill tasks assigned by your supervisor.\n 
-        [Github](https://github.com/talkingwallace/ChatGPT-Paper-Reader)""")
-    
-
-app = GUI()
-def start_analyse(api_key, pdf_file):
-    return app.analyse(api_key, pdf_file)
-
-def start_ask(question):
-    return app.ask_question(question)
-
-iface = gr.Interface(fn=start_analyse,inputs=["text", "file"],outputs="text",title=title,description=description)
-upload_button = iface.get_widget("file")
-upload_button.label = "Upload PDF File"
-iface.inputs[1].label = "Upload PDF File"  # Set the label for the file input
-
-question_input = gr.inputs.Textbox(label="Your Question", placeholder="Authors of this paper?")
-ask_button = gr.Button("Ask")
+        """
+        # CHATGPT-PAPER-READER
+        """)
+    gr.HTML(title)
+    gr.HTML(description)
+    with gr.Tab("Upload PDF File"):
+        pdf_input = gr.File(label="PDF File")
+        api_input = gr.Textbox(label="OpenAI API Key")
+        result = gr.Textbox(label="PDF Summary")
+        upload_button = gr.Button("Start Analyse")
+    with gr.Tab("Ask question about your PDF"):
+        question_input = gr.Textbox(label="Your Question", placeholder="Authors of this paper?")
+        answer = gr.Textbox(label="Answer")
+        ask_button = gr.Button("Ask")
+    with gr.Accordion("About this project"):
+        gr.Markdown(
+            """## CHATGPT-PAPER-READER📝 
+            This repository provides a simple interface that utilizes the gpt-3.5-turbo 
+            model to read academic papers in PDF format locally. You can use it to help you summarize papers, 
+            create presentation slides, or simply fulfill tasks assigned by your supervisor.\n 
+            [Github](https://github.com/talkingwallace/ChatGPT-Paper-Reader)""")
 
-iface.add_input(question_input)
-iface.add_output(ask_button)
+    app = GUI()
+    upload_button.click(fn=app.analyse, inputs=[api_input, pdf_input], outputs=result)
+    ask_button.click(app.ask_question, inputs=question_input, outputs=answer)
 
-iface.launch()
+if __name__ == "__main__":
+    demo.title = "CHATGPT-PAPER-READER"
+    demo.launch()