Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 # Initialize the Hugging Face Inference Client
-client = InferenceClient()
+client = InferenceClient(api_key="YOUR_HF_TOKEN")
 
-# Function to interact with the Qwen2.5-32B model and analyze code for compliance
-def analyze_compliance(code, compliance_standard):
+# Function to stream the compliance suggestions as they are generated
+def analyze_compliance_stream(code, compliance_standard):
     prompt = f"Analyze the following code for {compliance_standard} compliance and suggest modifications or refactoring to meet the guidelines:\n\n{code}"
     
     messages = [
@@ -22,29 +22,35 @@ def analyze_compliance(code, compliance_standard):
         stream=True
     )
 
-    # Collect the output from the stream
+    # Stream content as it is generated
     compliance_suggestions = ""
     for chunk in stream:
         compliance_suggestions += chunk.choices[0].delta.content
+        yield compliance_suggestions  # Yield incremental content to display immediately
 
-    return compliance_suggestions
-
-# Create Gradio interface
+# Create Gradio interface with the modified layout
 with gr.Blocks() as app:
-    gr.Markdown("## Legal and Compliance Code Advisor")
-    gr.Markdown("Analyze your code for legal compliance standards (e.g., GDPR, HIPAA) and receive modification suggestions.")
-
-    with gr.Row():
-        code_input = gr.Textbox(lines=10, label="Code Snippet", placeholder="Enter your code here")
-        compliance_standard = gr.Dropdown(
-            choices=["GDPR", "HIPAA", "PCI-DSS", "SOC 2", "ISO 27001"],
-            label="Compliance Standard",
-            value="GDPR"
-        )
-        output_text = gr.Textbox(label="Compliance Suggestions", placeholder="Compliance suggestions will appear here...")
+    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")
-    analyze_button.click(fn=analyze_compliance, inputs=[code_input, compliance_standard], outputs=output_text)
+
+    # Result display in Markdown to HTML
+    output_html = gr.HTML(label="Compliance Suggestions")
+
+    # Link button to function with inputs and outputs
+    analyze_button.click(fn=analyze_compliance_stream, inputs=[code_input, compliance_standard], outputs=output_html, every=0.1)
 
 # Run the Gradio app
-app.launch()
+app.launch()