Create app.py

app.py

@@ -0,0 +1,59 @@
+import gradio as gr
+import random
+from scraperr import scrape_google
+from pdf_converter import DocumentHandler
+
+
+
+async def scrape_websites(topic, num_links, num_results_per_link=10):
+    output_text = await scrape_google(topic, num_results_per_link)
+    # Ensure output_text is a list
+    if not isinstance(output_text, list):
+        output_text = [output_text]
+    # Select random links based on the user's input
+    selected_links = random.sample(output_text, min(num_links, len(output_text)))
+    # Convert the list of strings to a single string
+    output_string = "\n".join(selected_links)
+    return output_string
+
+def convert_to_pdf(url):
+    document_handler = DocumentHandler(url)
+    pdf_path = document_handler.create_pdf()
+    return pdf_path
+
+html = """
+<div style="text-align:center; max-width: 900px; margin: 0 auto; margin-top:5px">
+    <h1>Research Assistant</h1>
+    <p> Welcome to the Research Assistant app! This tool helps you find relevant information on your topic of interest.</p>
+</div>"""
+css = """container{max-width:900px; margin-left:auto; margin-right:auto;padding:20px}"""
+
+with gr.Blocks(css=css,theme=gr.themes.Monochrome( primary_hue= gr.themes.colors.orange , secondary_hue=gr.themes.colors.blue,neutral_hue=gr.themes.colors.gray,radius_size=gr.themes.sizes.radius_md)) as demo:
+    gr.HTML(html)
+    with gr.Tab("Scrape Google"):
+      with gr.Row():  
+            with gr.Column(variant="panel"):
+                
+                topic = gr.Textbox(label="What is your Research Topic?", container= True)
+                num_links = gr.Slider(label="Specify the Number of Links to Scrape!",minimum= 0, container= True)
+                text_button = gr.Button("Scrape")
+                
+            with gr.Column(min_width=600):
+                text_output = gr.Textbox(label="Result", lines= 10, container=True, autoscroll=False)
+        
+            text_button.click(scrape_websites, inputs=[topic, num_links], outputs=text_output, scroll_to_output=False)        
+            
+            
+    
+    with gr.Tab("Convert To PDF"):
+        url_input = gr.Textbox(label="URL")
+        pdf_button = gr.Button("Convert")
+        pdf_output = gr.File(label="PDF")
+        pdf_button.click(convert_to_pdf, inputs=url_input, outputs=pdf_output)
+        chat_bot_button = gr.Button("Proceed to ChatBot", link="https://huggingface.co/spaces/lara1510/chatbot")
+        
+        
+        
+
+if __name__ == "__main__":
+    demo.launch()