Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Initialize the Hugging Face Inference Client (you can set the api_key directly when you initialize the client)
+# Initialize the Hugging Face Inference Client
 client = InferenceClient()
 
 # Function to stream the compliance suggestions as they are generated
@@ -33,24 +33,24 @@ with gr.Blocks() as app:
     gr.Markdown("## Code Compliance Advisor")
     gr.Markdown("Analyze your code for legal compliance and security standards (e.g., GDPR, HIPAA) and receive actionable suggestions.")
 
-    # Full-width text area for code input
-    code_input = gr.Textbox(lines=10, label="Code Snippet", placeholder="Enter your code here", elem_id="full_width")
-
-    # Dropdown for selecting compliance standard
-    compliance_standard = gr.Dropdown(
-        choices=["GDPR", "HIPAA", "PCI-DSS", "SOC 2", "ISO 27001"],
-        label="Compliance Standard",
-        value="GDPR"
-    )
-
-    # Button to trigger analysis
-    analyze_button = gr.Button("Analyze Compliance")
-
-    # Result display in Markdown to HTML
-    output_html = gr.HTML(label="Compliance Suggestions")
+    with gr.Row():
+        # First column for input components
+        with gr.Column():
+            code_input = gr.Textbox(lines=10, label="Code Snippet", placeholder="Enter your code here", elem_id="full_width")
+            compliance_standard = gr.Dropdown(
+                choices=["GDPR", "HIPAA", "PCI-DSS", "SOC 2", "ISO 27001"],
+                label="Compliance Standard",
+                value="GDPR"
+            )
+            analyze_button = gr.Button("Analyze Compliance")
+
+        # Second column for output
+        with gr.Column():
+            gr.Markdown("### Compliance Suggestions")  # This acts as the label for the output
+            output_markdown = gr.Markdown()
 
     # Link button to function with inputs and outputs
-    analyze_button.click(fn=analyze_compliance_stream, inputs=[code_input, compliance_standard], outputs=output_html)
+    analyze_button.click(fn=analyze_compliance_stream, inputs=[code_input, compliance_standard], outputs=output_markdown)
 
 # Run the Gradio app
-app.launch()
+app.launch()