Update app.py

app.py

@@ -1,10 +1,8 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
-from fpdf import FPDF
 import torch
 import spaces
-import io
-import markdown
+
 
 # Initialize the Qwen model and tokenizer
 model_name = "Qwen/Qwen2.5-Coder-7B-Instruct"
@@ -30,36 +28,20 @@ def generate_documentation(code_input):
     generated_ids = [output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)]
     documentation = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
     print(documentation)
-    return markdown.markdown(documentation)
-
-# Function to generate and download PDF
-def create_pdf(documentation):
-    pdf = FPDF()
-    pdf.set_auto_page_break(auto=True, margin=15)
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
-    pdf.multi_cell(200, 10, documentation)
+    return documentation
 
-    # Create a bytes buffer and output the PDF to it
-    buffer = io.BytesIO()
-    pdf.output(buffer)
-    buffer.seek(0)
-    
-    return buffer
 
 # Gradio interface
 def process_code(code_input):
     documentation = generate_documentation(code_input)
-    pdf_path = create_pdf(documentation)
-    return documentation, pdf_path
+    return documentation
 
 # Set up the Gradio app with Bootstrap, icons, and smiley
 with gr.Blocks(css=".container { font-family: 'Roboto', sans-serif; } .btn-primary { background-color: #007bff; } .icon { margin-right: 10px; }") as app:
     gr.Markdown("""
     #Code Documentation Generator
     
-    Paste your code below, and the app will generate the README and detailed documentation for you.
-    The output will also be available for download as a PDF.
+    Paste your code below, and the app will generate the Documentation for you.
     """)
     
     with gr.Row():
@@ -69,11 +51,10 @@ with gr.Blocks(css=".container { font-family: 'Roboto', sans-serif; } .btn-prima
         generate_button = gr.Button("Generate Documentation", elem_classes="btn btn-primary")
     
     with gr.Row():
-        output_text = gr.Textbox(label="Generated Documentation", lines=20, interactive=False)
-        download_pdf = gr.File(label="Download PDF", file_types=[".pdf"])
+        output_text = gr.Markdown()
     
     # Bind function to button click
-    generate_button.click(process_code, inputs=code_input, outputs=[output_text, download_pdf])
+    generate_button.click(process_code, inputs=code_input, outputs=output_text)
 
 # Launch the Gradio app
 app.launch()