✨ Feature: Add Auto-Pull Request creation via GitHub API directly from Streamlit UI

app.py

@@ -10,7 +10,7 @@ from typing import Optional
 import streamlit as st
 
 from backend.agent import AgentResult, FixFlowAgent, generate_full_report
-from backend.config import GLM_MODEL, GLM_BASE_URL
+from backend.config import GLM_MODEL, GLM_BASE_URL, GLM_API_KEY, GITHUB_TOKEN
 from backend.github_client import GitHubClient
 from backend.llm_client import GLMClient
 
@@ -388,8 +388,8 @@ def init_session():
         "step_messages": {},
         "stream_buffer": "",
         "error": None,
-        "glm_api_key": "",
-        "github_token": "",
+        "glm_api_key": GLM_API_KEY,
+        "github_token": GITHUB_TOKEN,
         "model": GLM_MODEL,
         "run_confidence": False,
     }
@@ -433,7 +433,7 @@ with st.sidebar:
 
     model_choice = st.selectbox(
         "GLM Model",
-        options=["glm-5-plus", "glm-4-plus", "glm-4"],
+        options=["glm-5", "glm-5.1", "glm-5-plus", "glm-4-plus", "glm-4"],
         index=0,
         key="model_select",
     )
@@ -517,7 +517,6 @@ if issue_url and not repo_url:
     import re
     m = re.match(r"(https://github\.com/[^/]+/[^/]+)/issues/\d+", issue_url.strip())
     if m:
-        st.session_state["repo_url_input"] = m.group(1)
         repo_url = m.group(1)
 
 # Example buttons
@@ -786,15 +785,36 @@ if st.session_state.result:
         # Copy button for full diff
         if result.diff_formatted and result.diffs:
             st.markdown("---")
-            copy_col, _ = st.columns([1, 3])
+            copy_col, pr_col, _ = st.columns([1, 1, 2])
             with copy_col:
                 st.download_button(
-                    "📋 Copy Full Diff",
+                    "📋 Download .diff Patch",
                     data=result.diff_formatted,
                     file_name="fixflow.diff",
                     mime="text/plain",
                     use_container_width=True,
                 )
+            with pr_col:
+                if st.button("🚀 Open GitHub Pull Request", use_container_width=True, type="primary"):
+                    if not st.session_state.github_token:
+                        st.error("⚠️ A GitHub Token with write access is required to open a PR.")
+                    else:
+                        with st.spinner("🚀 Creating Pull Request..."):
+                            gh = GitHubClient(token=st.session_state.github_token)
+                            try:
+                                branch_name = f"fixflow-patch-{int(time.time())}"
+                                title = f"Fix: {result.issue_data.get('title', 'Issue')}"
+                                body = result.fix_explanation + "\n\n---\n*Generated autonomously by FixFlow*"
+                                pr_url = gh.create_pull_request(
+                                    repo_url=result.repo_url,
+                                    branch_name=branch_name,
+                                    files_content=result.fixed_files,
+                                    title=title,
+                                    body=body
+                                )
+                                st.success(f"✅ Created successfully: [View PR]({pr_url})")
+                            except Exception as e:
+                                st.error(f"Failed to create PR: {e}")
 
     # ── Step 5: Fix Explanation ───────────────────────────────────────────────
     with st.expander("📝 Step 5: PR Description & Fix Explanation", expanded=True):

backend/config.py

@@ -14,6 +14,8 @@ GLM_MODEL: str = os.getenv("GLM_MODEL", "glm-5-plus")
 
 # ── GitHub Config ────────────────────────────────────────────────────────────
 GITHUB_TOKEN: str = os.getenv("GITHUB_TOKEN", "")
+if GITHUB_TOKEN == "your_github_token_here":
+    GITHUB_TOKEN = ""
 
 # ── Agent Limits ─────────────────────────────────────────────────────────────
 MAX_FILES_TO_SCAN: int = int(os.getenv("MAX_FILES_TO_SCAN", "100"))

backend/github_client.py

@@ -228,6 +228,68 @@ class GitHubClient:
                 result[path] = content
         return result
 
+    # ── Pull Request Creation ─────────────────────────────────────────────────
+
+    def create_pull_request(
+        self,
+        repo_url: str,
+        branch_name: str,
+        files_content: Dict[str, str],
+        title: str,
+        body: str,
+    ) -> str:
+        """
+        Creates a new branch and commits all changed files, then opens a pull request.
+        Requires a GitHub token with write access to the repository.
+        Returns the HTML URL of the created PR.
+        """
+        if not self._gh.get_user():
+            raise RuntimeError("A valid GitHub Token with write access is required to create a PR.")
+            
+        owner, repo_name = parse_repo_url(repo_url)
+        logger.info("Creating PR on %s/%s branch %s", owner, repo_name, branch_name)
+        
+        try:
+            repo = self._gh.get_repo(f"{owner}/{repo_name}")
+            from github import InputGitTreeElement
+            
+            base_branch = repo.default_branch
+            base_ref = repo.get_git_ref(f"heads/{base_branch}")
+            
+            # Create new branch off base branch
+            try:
+                repo.create_git_ref(ref=f"refs/heads/{branch_name}", sha=base_ref.object.sha)
+            except GithubException:
+                logger.warning(f"Branch {branch_name} may already exist, proceeding to update it.")
+            
+            base_tree = repo.get_git_tree(base_ref.object.sha)
+            
+            # Create a blob for each changed file
+            elements = []
+            for filepath, content in files_content.items():
+                blob = repo.create_git_blob(content, "utf-8")
+                elements.append(
+                    InputGitTreeElement(path=filepath, mode='100644', type='blob', sha=blob.sha)
+                )
+                
+            # Create new tree with all blob changes batched together
+            new_tree = repo.create_git_tree(elements, base_tree)
+            parent = repo.get_git_commit(base_ref.object.sha)
+            commit = repo.create_git_commit(message=title, tree=new_tree, parents=[parent])
+            
+            # Update the branch reference to point to the new commit
+            ref = repo.get_git_ref(f"heads/{branch_name}")
+            ref.edit(commit.sha)
+            
+            # Create the actual PR
+            pr = repo.create_pull(title=title, body=body, head=branch_name, base=base_branch)
+            return pr.html_url
+            
+        except GithubException as e:
+            raise RuntimeError(
+                f"Failed to create PR. Ensure your GitHub token has write access to {owner}/{repo_name}. Detail: {e.data.get('message', str(e))}"
+            ) from e
+
     # ── Rate Limit Info ───────────────────────────────────────────────────────
 
     def get_rate_limit_info(self) -> Dict: