Create app.py

app.py

@@ -0,0 +1,100 @@
+import PyPDF2
+import gradio as gr
+import openai
+from huggingface_hub import HfApi
+
+api = HfApi()
+openai.api_key = api.get_secret("OPENAI_API_KEY")
+
+def extract_text_from_block(pdf_path, block_pages):
+    text = ""
+    with open(pdf_path, 'rb') as file:
+        reader = PyPDF2.PdfReader(file)
+        for page_num in range(block_pages[0] - 1, block_pages[1]):
+            page = reader.pages[page_num]
+            text += page.extract_text()
+    return text
+
+def generate_test_cases(text, prompt_template):
+    prompt = prompt_template.format(text=text)
+    response = openai.ChatCompletion.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant that generates test cases based on given text."},
+            {"role": "user", "content": prompt}
+        ],
+        max_tokens=1000,
+        n=1,
+        temperature=0.7,
+    )
+    return response.choices[0].message['content'].strip()
+
+def process_pdf(pdf_file, selected_block, prompt_template):
+    if pdf_file is None:
+        return "Please upload a PDF file."
+    
+    block_config = {
+        "5.4.1 OSPI-xSPI-QSPI-SPI Boot": (482, 488),
+        "5.4.2 I2C Boot": (489, 489),
+        "5.4.3 SD Card Boot": (490, 491),
+        "5.4.4 eMMC Boot": (492, 493),
+        "5.4.5 Ethernet Boot": (494, 497),
+        "5.4.6 USB Boot": (498, 498),
+        "5.4.7 UART Boot": (499, 500),
+        "5.4.8 GPMC NOR Boot": (501, 502),
+        "5.4.9 GPMC NAND Boot": (503, 503),
+        "5.4.10 Serial NAND Boot": (504, 505),
+        "5.4.11 No boot/Development boot": (506, 506)
+    }
+
+    try:
+        temp_pdf_path = pdf_file.name
+        selected_pages = block_config[selected_block]
+        
+        text = extract_text_from_block(temp_pdf_path, selected_pages)
+        test_cases = generate_test_cases(text, prompt_template)
+
+        return f"Generated Test Cases for {selected_block}:\n\n{test_cases}"
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
+block_names = [
+    "5.4.1 OSPI-xSPI-QSPI-SPI Boot",
+    "5.4.2 I2C Boot",
+    "5.4.3 SD Card Boot",
+    "5.4.4 eMMC Boot",
+    "5.4.5 Ethernet Boot",
+    "5.4.6 USB Boot",
+    "5.4.7 UART Boot",
+    "5.4.8 GPMC NOR Boot",
+    "5.4.9 GPMC NAND Boot",
+    "5.4.10 Serial NAND Boot",
+    "5.4.11 No boot/Development boot"
+]
+
+default_prompt = """Generate test cases for the following text:
+
+{text}
+
+Please structure each test case as follows:
+- Test ID
+- Test Summary
+- Pre-requisite
+- Steps
+- Expected Result
+
+Test Cases:"""
+
+app = gr.Interface(
+    fn=process_pdf,
+    inputs=[
+        gr.File(label="Upload PDF"),
+        gr.Dropdown(choices=block_names, label="Select Block"),
+        gr.Textbox(label="Prompt Template", value=default_prompt, lines=10)
+    ],
+    outputs="text",
+    title="PDF Block Test Case Generator",
+    description="Upload a PDF, select a block, edit the prompt if needed, and generate test cases for that block."
+)
+
+app.launch()